fix mypy in test_array.py

brokkoli71 · brokkoli71 · commit 995a5d58c46c · 2025-02-15T21:27:33.000+01:00
diff --git a/tests/test_array.py b/tests/test_array.py
@@ -1463,4 +1463,6 @@ async def test_sharding_coordinate_selection() -> None:
         shards=(2, 4, 4),
     )
     arr[:] = np.arange(2 * 3 * 4).reshape((2, 3, 4))
-    assert (arr[1, [0, 1]] == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all()  # type: ignore[index]
+    result = arr[1, [0, 1]]
+    assert isinstance(result, NDArrayLike)
+    assert (result == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all()  # type: ignore[index]

Original file line number	Diff line number	Diff line change
`@@ -1463,4 +1463,6 @@ async def test_sharding_coordinate_selection() -> None:`
`1463`	`1463`	`shards=(2, 4, 4),`
`1464`	`1464`	`)`
`1465`	`1465`	`arr[:] = np.arange(2 * 3 * 4).reshape((2, 3, 4))`
`1466`		`- assert (arr[1, [0, 1]] == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all() # type: ignore[index]`
	`1466`	`+ result = arr[1, [0, 1]]`
	`1467`	`+ assert isinstance(result, NDArrayLike)`
	`1468`	`+ assert (result == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all() # type: ignore[index]`