fix veg frequency test

opendatacube · Dec 5, 2024 · b72dcb7 · b72dcb7
1 parent 51e432b
commit b72dcb7
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/tests/test_landcover_plugin_a0.py b/tests/test_landcover_plugin_a0.py
@@ -400,14 +400,15 @@ def test_reduce(fc_wo_dataset):
     xx = stats_veg.reduce(xx).compute()
     expected_value = np.array(
         [
-            [1, 255, 0, 255, 255, 255, 255],
-            [1, 255, 255, 255, 255, 255, 255],
-            [255, 0, 255, 255, 1, 255, 255],
-            [255, 255, 1, 255, 255, 255, 255],
+            [2, 255, 255, 255, 255, 255, 255],
+            [2, 255, 255, 255, 255, 255, 255],
+            [255, 255, 255, 255, 2, 255, 255],
+            [255, 255, 2, 255, 255, 255, 255],
             [255, 255, 255, 255, 255, 255, 255],
-            [255, 1, 255, 255, 255, 255, 255],
-            [255, 255, 255, 255, 255, 255, 0],
-        ]
+            [255, 2, 255, 255, 255, 255, 255],
+            [255, 255, 255, 255, 255, 255, 255],
+        ],
+        dtype="uint8",
     )
 
     assert (xx.veg_frequency.data == expected_value).all()