TST: Solve compatibility with pytest 7.1

pytest.approx no longer wants to work with unordered sequences
equinor · Mar 14, 2022 · 0bed984 · 0bed984
1 parent b5ff572
commit 0bed984
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/test_surface/test_file_io.py b/tests/test_surface/test_file_io.py
@@ -167,7 +167,9 @@ def test_read_cube(data):
     }
     cube = Cube(**cube_input)
     surf_from_cube = RegularSurface._read_cube(cube, data)
-    assert set(surf_from_cube.values.data.flatten().tolist()) == pytest.approx({data})
+    assert list(set(surf_from_cube.values.data.flatten().tolist())) == pytest.approx(
+        [data]
+    )
 
     cube_input.pop("zinc")
     cube_input.pop("nlay")

diff --git a/tests/test_surface/test_new_surface.py b/tests/test_surface/test_new_surface.py
@@ -88,7 +88,7 @@ def test_wrong_size_input_lists(data):
 @given(data=st.one_of(st.floats(allow_nan=False), st.integers()))
 def test_input_numbers(data):
     surf = RegularSurface(10, 10, 0.0, 0.0, values=data)
-    assert set(surf.values.data.flatten().tolist()) == pytest.approx({data})
+    assert list(set(surf.values.data.flatten().tolist())) == pytest.approx([data])
 
 
 def test_read_grid3d():