From 7a5aa06d51550b636897363a7141681dd18ec494 Mon Sep 17 00:00:00 2001 From: Pierrick Rambaud Date: Sat, 22 Feb 2025 08:26:08 +0000 Subject: [PATCH] test: use a pytest fixture --- tests/conftest.py | 15 +++++++++++++++ tests/test_Image.py | 28 ++++------------------------ 2 files changed, 19 insertions(+), 24 deletions(-) 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: