delineator/tests/conftest.py at 699376f4ad5a985936672a21c0b8ca47804bf3d6 · Upstream-Tech/delineator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
"""
Pytest configuration and fixtures for watershed delineation tests.

This module sets up the environment variables needed to access remote hydrology data
and provides common fixtures used across test modules.
"""

import os
import tempfile
from pathlib import Path

import pytest

# Set environment variables at module load time, before any imports
# These are required by upstream_delineator modules at import time
os.environ.setdefault(
    "CATCHMENT_PATH", "https://public-hydrology-data.upstream.tech/catchments"
)
os.environ.setdefault(
    "RIVER_PATH", "https://public-hydrology-data.upstream.tech/rivers"
)
os.environ.setdefault(
    "FLOW_DIR_PATH", "https://public-hydrology-data.upstream.tech/merit_flowdir.tif"
)
os.environ.setdefault(
    "ACCUM_PATH", "https://public-hydrology-data.upstream.tech/merit_accum.tif"
)
os.environ.setdefault(
    "MEGABASINS_PATH",
    "https://public-hydrology-data.upstream.tech/merit_basins_lvl2.gpkg",
)


@pytest.fixture(scope="session")
def temp_output_dir():
    """Create a temporary directory for test outputs."""
    with tempfile.TemporaryDirectory() as tmpdir:
        yield Path(tmpdir)


@pytest.fixture
def single_outlet_csv(tmp_path):
    """
    Create a CSV file with a single outlet point.
    Uses a location in Iceland (megabasin 27) for testing.
    """
    csv_path = tmp_path / "single_outlet.csv"
    csv_path.write_text(
        "id,lng,lat,name,outlet_id\n"
        "outlet1,-14.36201,65.50253,Lagarfljot River at Lagarfoss,outlet1\n"
    )
    return str(csv_path)


@pytest.fixture
def multi_subbasin_csv(tmp_path):
    """
    Create a CSV file with an outlet and upstream subbasin points.
    This tests the subbasin delineation capability.
    """
    csv_path = tmp_path / "multi_subbasin.csv"
    csv_path.write_text(
        "id,lng,lat,name,outlet_id\n"
        "main_outlet,-14.36201,65.50253,Lagarfljot River at Lagarfoss,main_outlet\n"
        "upstream1,-15.0883,64.9839,Jokulsa I River at Fljotsdal Holl,main_outlet\n"
        "upstream2,-14.533,65.14,Gringa Dam,main_outlet\n"
    )
    return str(csv_path)


@pytest.fixture
def headwater_outlet_csv(tmp_path):
    """
    Create a CSV file with a headwater outlet point.
    Headwater points are typically in leaf catchments with no upstream neighbors.
    """
    csv_path = tmp_path / "headwater.csv"
    # A point near the headwaters of a small stream in Iceland
    csv_path.write_text(
        "id,lng,lat,name,outlet_id\n"
        "headwater,-15.186,64.735,Lake Sauoarvatr outlet,headwater\n"
    )
    return str(csv_path)


@pytest.fixture
def default_config():
    """Default configuration for tests - minimal output, no plots."""
    return {
        "VERBOSE": False,
        "WRITE_OUTPUT": False,
        "PLOTS": False,
        "CONSOLIDATE": False,
        "NETWORK_DIAGRAMS": False,
        "SIMPLIFY": False,
    }


@pytest.fixture
def consolidate_config():
    """Configuration with consolidation enabled."""
    return {
        "VERBOSE": False,
        "WRITE_OUTPUT": False,
        "PLOTS": False,
        "CONSOLIDATE": True,
        "MAX_AREA": 500,
        "NETWORK_DIAGRAMS": False,
        "SIMPLIFY": False,
    }