add a better test

LDeakin · LDeakin · commit 55ebb3104e7e · 2025-02-12T08:36:40.000+11:00
diff --git a/tests/test_array.py b/tests/test_array.py
@@ -1433,5 +1433,5 @@ async def test_sharding_coordinate_selection() -> None:
         dtype=np.float32,
         shards=(5, 20, 30),
     )
-    arr[:] = 1
-    assert (arr[5, [0, 1]] == 1).all()
+    arr[:] = np.arange(10*20*30).reshape((10, 20, 30))
+    assert (arr[5, [0, 1]] == np.vstack([np.arange(3000, 3030), np.arange(3030, 3060)])).all()

Original file line number	Diff line number	Diff line change
`@@ -1433,5 +1433,5 @@ async def test_sharding_coordinate_selection() -> None:`
`1433`	`1433`	`dtype=np.float32,`
`1434`	`1434`	`shards=(5, 20, 30),`
`1435`	`1435`	`)`
`1436`		`- arr[:] = 1`
`1437`		`- assert (arr[5, [0, 1]] == 1).all()`
	`1436`	`+ arr[:] = np.arange(102030).reshape((10, 20, 30))`
	`1437`	`+ assert (arr[5, [0, 1]] == np.vstack([np.arange(3000, 3030), np.arange(3030, 3060)])).all()`