fix mypy in test_array.py

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

Original file line number	Diff line number	Diff line change
`@@ -1463,6 +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`		`- result = arr[1, [0, 1]]`
	`1466`	`+ result = arr[1, [0, 1]] # type: ignore[index]`
`1467`	`1467`	`assert isinstance(result, NDArrayLike)`
`1468`		`- assert (result == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all() # type: ignore[index]`
	`1468`	`+ assert (result == np.array([[12, 13, 14, 15], [16, 17, 18, 19]])).all()`