Skip to content

Commit 87698ea

Browse files
committed
Fix test (attempt 2)
1 parent 1868ac5 commit 87698ea

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

tests/test_1285_rntuple_multicluster_concatenation.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,7 @@ def test_schema_extension():
1414
arrays = obj.arrays()
1515
int_vec_array = arrays["int_vector"]
1616

17-
true_array = np.zeros((200, 2), dtype=np.int16)
18-
true_array[:100, 0] = np.arange(100, dtype=np.int16)
19-
true_array[:100, 1] = np.arange(100, dtype=np.int16)
20-
true_array[100:, 0] = np.arange(100, dtype=np.int16)
21-
true_array[100:, 1] = np.arange(100, dtype=np.int16) + 1
22-
23-
assert np.array_equal(int_vec_array, true_array)
17+
for j in range(2):
18+
for i in range(100):
19+
assert int_vec_array[i + j * 100, 0] == i
20+
assert int_vec_array[i + j * 100, 1] == i + j

0 commit comments

Comments
 (0)