Run main experiments

This commit is contained in:
Tobias Hölzer 2026-02-12 18:34:35 +01:00
parent 3ce6b6e867
commit f9df8e9fe6
43 changed files with 4112 additions and 4022 deletions

View file

@ -18,7 +18,6 @@ def sample_ensemble() -> Generator[DatasetEnsemble]:
grid="hex",
level=3, # Use level 3 for much faster tests
members=["AlphaEarth"], # Use only one member for faster tests
add_lonlat=True,
)
@ -29,7 +28,6 @@ def sample_ensemble_v2() -> Generator[DatasetEnsemble]:
grid="hex",
level=3, # Use level 3 for much faster tests
members=["AlphaEarth"], # Use only one member for faster tests
add_lonlat=True,
)
@ -41,7 +39,6 @@ class TestDatasetEnsemble:
assert sample_ensemble.grid == "hex"
assert sample_ensemble.level == 3
assert "AlphaEarth" in sample_ensemble.members
assert sample_ensemble.add_lonlat is True
def test_get_targets_returns_geodataframe(self, sample_ensemble: DatasetEnsemble) -> None:
"""Test that get_targets() returns a GeoDataFrame."""
@ -106,10 +103,9 @@ class TestDatasetEnsemble:
# Should NOT have geometry column
assert "geometry" not in features.columns
# Should have location columns if add_lonlat is True
if sample_ensemble.add_lonlat:
assert "x" in features.columns
assert "y" in features.columns
# Should have location columns
assert "x" in features.columns
assert "y" in features.columns
# Should have grid property columns
assert "cell_area" in features.columns