def test_iter_keys_3d(self):
row_arr = RowMajorArray(shape=(2, 2, 2))
assert list(row_arr.iter_keys()) == [(0, 0, 0), (0, 0, 1), (0, 1, 0),
(0, 1, 1), (1, 0, 0), (1, 0, 1),
(1, 1, 0), (1, 1, 1)]
row_arr = RowMajorArray(shape=(8, 8, 8))
assert len(list(row_arr.iter_keys())) == 8**3
def test_fill(self):
row_arr = RowMajorArray([[False, False], [False, False]])
for key in row_arr.iter_keys():
assert not row_arr[key]
row_arr.fill(True, dtype=bool)
for key in row_arr.iter_keys():
assert row_arr[key]
def test_iter_keys_2d(self):
row_arr = RowMajorArray(shape=(4, 4))
assert list(row_arr.iter_keys()) == [(0, 0), (0, 1), (0, 2), (0, 3),
(1, 0), (1, 1), (1, 2), (1, 3),
(2, 0), (2, 1), (2, 2), (2, 3),
(3, 0), (3, 1), (3, 2), (3, 3)]
row_arr = RowMajorArray(shape=(32, 32))
assert len(list(row_arr.iter_keys())) == 32**2