TST: test wuhan on nightly and push to main

.github/workflows/testing.yml

@@ -75,7 +75,24 @@ jobs:
           --cov-append \
           --cov-report term-missing \
           --cov-report xml . \
-          --env_type ${{ matrix.environment-file }}
+          --env_type ${{ matrix.environment-file }} \
+          -m "not wuhan"
+
+      - name: run tests (Wuhan)
+        run: |
+          pytest \
+          neatnet/ \
+          --verbose \
+          -r a \
+          --numprocesses logical \
+          --color yes \
+          --cov neatnet \
+          --cov-append \
+          --cov-report term-missing \
+          --cov-report xml . \
+          --env_type ${{ matrix.environment-file }} \
+          -m wuhan
+        if: ${{ github.event_name != 'pull_request' }}
 
       - name: zip artifacts - Ubuntu & macOS
         run: zip ci_artifacts.zip ci_artifacts -r

data/generate_simplified.py

@@ -27,12 +27,13 @@
 logging.info("")
 
 fua_city = {
-    1133: "aleppo",
-    869: "auckland",
-    4617: "bucaramanga",
-    809: "douala",
-    1656: "liege",
-    4881: "slc",
+    # 1133: "aleppo",
+    # 869: "auckland",
+    # 4617: "bucaramanga",
+    # 809: "douala",
+    # 1656: "liege",
+    # 4881: "slc",
+    8989: "wuhan",
 }
 
 # dict of cityname: fua ID

neatnet/tests/test_simplify.py

@@ -108,3 +108,29 @@ def test_simplify_network_full_fua(aoi, tol, known_length):
             observed.drop(columns=["_status", "geometry"]),
         )
         pytest.geom_test(known, observed, tolerance=tol, aoi=aoi)
+
+
+@pytest.mark.wuhan
+def test_simplify_network_wuhan(aoi="wuhan_8989", tol=0.3, known_length=4_702_861):
+    known = geopandas.read_parquet(full_fua_data / aoi / "simplified.parquet")
+    observed = neatnet.simplify_network(
+        geopandas.read_parquet(full_fua_data / aoi / "original.parquet")
+    )
+    observed_length = observed.geometry.length.sum()
+    assert "highway" in observed.columns
+
+    # storing GH artifacts
+    artifact_dir = ci_artifacts / aoi
+    artifact_dir.mkdir(parents=True, exist_ok=True)
+    observed.to_parquet(artifact_dir / "simplified.parquet")
+
+    assert pytest.approx(observed_length, rel=0.0001) == known_length
+    assert observed.index.dtype == numpy.dtype("int64")
+
+    if pytest.ubuntu and pytest.env_type != "oldest":
+        assert_series_equal(known["_status"], observed["_status"])
+        assert_frame_equal(
+            known.drop(columns=["_status", "geometry"]),
+            observed.drop(columns=["_status", "geometry"]),
+        )
+        pytest.geom_test(known, observed, tolerance=tol, aoi=aoi)

pyproject.toml

@@ -87,6 +87,9 @@ filterwarnings = [
     # this is an internal warning thrown within ``neatnet.geometry.snap_to_targets()``
     'ignore:Could not create a connection*:UserWarning',
 ]
+markers = [
+    'wuhan:Wuhan takes ages to run'
+]
 [tool.pixi.project]
 channels = ["conda-forge"]
 platforms = ["linux-64", "osx-arm64", "osx-64", "win-64"]