diff --git a/tests/conftest.py b/tests/conftest.py index a90bf425..a8bf3480 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -275,3 +275,18 @@ def s2_class_image(): This image is located in Argentina and Chile (Patagonia). """ return ee.Image("COPERNICUS/S2_SR_HARMONIZED/20230120T142709_20230120T143451_T18GYT") + + +@pytest.fixture +def polygon_instance(): + """Return a defined polygon instance.""" + return ee.Geometry.Polygon( + [ + [ + [-71.84476689765019, -42.81816243454466], + [-71.84476689765019, -42.897690198549135], + [-71.72391728827519, -42.897690198549135], + [-71.72391728827519, -42.81816243454466], + ] + ] + ) diff --git a/tests/test_Image.py b/tests/test_Image.py index a15f6b80..61d9a126 100644 --- a/tests/test_Image.py +++ b/tests/test_Image.py @@ -621,27 +621,17 @@ def image(self): class TestClassToBands: """Test the ``classToBands`` method.""" - def test_class_to_bands(self, s2_class_image, ee_image_regression): + def test_class_to_bands(self, s2_class_image, polygon_instance, ee_image_regression): """Test the ``classToBands`` method.""" class_info = {"3": "shadow", "9": "clouds_high", "11": "snow"} decoded = s2_class_image.geetools.classToBands(class_info, "SCL") - aoi = ee.Geometry.Polygon( - [ - [ - [-71.84476689765019, -42.81816243454466], - [-71.84476689765019, -42.897690198549135], - [-71.72391728827519, -42.897690198549135], - [-71.72391728827519, -42.81816243454466], - ] - ] - ) - ee_image_regression.check(decoded, scale=10, region=aoi) + ee_image_regression.check(decoded, scale=10, region=polygon_instance) class TestClassMask: """Test the ``classMask`` method.""" - def test_class_mask(self, s2_class_image, ee_image_regression): + def test_class_mask(self, s2_class_image, polygon_instance, ee_image_regression): """Test the ``classMask`` method.""" class_info = { "2": "dark", @@ -652,17 +642,7 @@ def test_class_mask(self, s2_class_image, ee_image_regression): "10": "cirrus", } decoded = s2_class_image.geetools.classMask(class_info, band="SCL") - aoi = ee.Geometry.Polygon( - [ - [ - [-71.84476689765019, -42.81816243454466], - [-71.84476689765019, -42.897690198549135], - [-71.72391728827519, -42.897690198549135], - [-71.72391728827519, -42.81816243454466], - ] - ] - ) - ee_image_regression.check(decoded, scale=10, region=aoi) + ee_image_regression.check(decoded, scale=10, region=polygon_instance) class TestPlot: