Run main experiments
This commit is contained in:
parent
3ce6b6e867
commit
f9df8e9fe6
43 changed files with 4112 additions and 4022 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue