use sets for dimension checks

Open-EO · Nov 3, 2023 · 9e98067 · 9e98067
1 parent 8a8cb06
commit 9e98067
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_mogpr.py b/tests/test_mogpr.py
@@ -24,7 +24,7 @@ def test_mogpr_udf(data):
     from fusets.openeo.mogpr_udf import apply_datacube
 
     result = apply_datacube(XarrayDataCube(data.to_array(dim="bands")), context={})
-    assert result.array.dims == ("bands", "t", "y", "x")
+    assert set(result.array.dims) == {"bands", "t", "y", "x"}
     assert result.array.shape == (2, 146, 2, 2)
 
 
@@ -33,5 +33,5 @@ def test_mogpr_train_model(data):
     t.fit(data)
     out = t.transform(data)
 
-    assert tuple(out.coords) == ("t", "y", "x")
+    assert set(out.coords) == {"t", "y", "x"}
     assert out.NDVI.shape == (146, 2, 2)