chore: fix tests for py3.12

ZettaAI · Feb 4, 2025 · bf0881e · bf0881e
1 parent 395790e
commit bf0881e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/tests/unit/layer/test_layer_set.py b/tests/unit/layer/test_layer_set.py
@@ -10,8 +10,8 @@ def test_read(mocker):
     idx = mocker.MagicMock()
     result = layer_set[idx]
     assert result == {"a": "a", "b": "b"}
-    layer_a.read_with_procs.called_with(idx=idx)
-    layer_b.read_with_procs.called_with(idx=idx)
+    layer_a.read_with_procs.assert_called_with(idx)
+    layer_b.read_with_procs.assert_called_with(idx)
 
 
 def test_write(mocker):
@@ -22,5 +22,5 @@ def test_write(mocker):
     layer_set = build_layer_set(layers={"a": layer_a, "b": layer_b})
     idx = mocker.MagicMock()
     layer_set[idx] = {"a": 1, "b": 2}
-    layer_a.write_with_procs.called_with(idx=idx, data=1)
-    layer_b.write_with_procs.called_with(idx=idx, data=2)
+    layer_a.write_with_procs.assert_called_with(idx, 1)
+    layer_b.write_with_procs.assert_called_with(idx, 2)
diff --git a/tests/unit/layer/volumetric/test_layer_set.py b/tests/unit/layer/volumetric/test_layer_set.py
@@ -13,8 +13,8 @@ def test_read(mocker):
     idx = VolumetricIndex(bbox=BBox3D(bounds=((0, 1), (0, 1), (0, 1))), resolution=Vec3D(1, 1, 1))
     result = layer_set[idx]
     assert result == {"a": np.array([1]), "b": np.array([2])}
-    layer_a.read_with_procs.called_with(idx=idx)
-    layer_b.read_with_procs.called_with(idx=idx)
+    layer_a.read_with_procs.assert_called_with(idx)
+    layer_b.read_with_procs.assert_called_with(idx)
 
 
 def test_write(mocker):
@@ -25,5 +25,5 @@ def test_write(mocker):
     layer_set = build_volumetric_layer_set(layers={"a": layer_a, "b": layer_b})
     idx = VolumetricIndex(bbox=BBox3D(bounds=((0, 1), (0, 1), (0, 1))), resolution=Vec3D(1, 1, 1))
     layer_set[idx] = {"a": 1, "b": 2}
-    layer_a.write_with_procs.called_with(idx=idx, data=1)
-    layer_b.write_with_procs.called_with(idx=idx, data=2)
+    layer_a.write_with_procs.assert_called_with(idx, 1)
+    layer_b.write_with_procs.assert_called_with(idx, 2)
diff --git a/zetta_utils/tmp.py b/zetta_utils/tmp.py