test full shape

Author: Matic Lubej

tests/mask/test_snow_mask.py

@@ -41,4 +41,4 @@ def test_snow_empty_eopatch(test_eopatch):
     empty_eopatch = EOPatch(bbox=test_eopatch.bbox, timestamps=[], data={"BANDS-S2-L1C": empty_bands_array})
     task = SnowMaskTask((FeatureType.DATA, "BANDS-S2-L1C"), [2, 3, 7, 11], mask_name="TEST_SNOW_MASK")
     resulting_eopatch = task(empty_eopatch)  # checks if the task runs without errors
-    assert resulting_eopatch.mask["TEST_SNOW_MASK"].shape[0] == 0
+    assert resulting_eopatch.mask["TEST_SNOW_MASK"].shape == (*empty_bands_array.shape[:-1], 1)