Skip to content

Commit

Permalink
Add test for flip in test_mem_utilization
Browse files Browse the repository at this point in the history
  • Loading branch information
tomwhite committed Aug 30, 2024
1 parent 11cd529 commit b1cf324
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions cubed/tests/test_mem_utilization.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,6 +243,16 @@ def test_concat(tmp_path, spec, executor):
run_operation(tmp_path, executor, "concat", c)


@pytest.mark.slow
def test_flip(tmp_path, spec, executor):
# Note 'a' has one fewer element in axis=0 to force chunking to cross array boundaries
a = cubed.random.random(
(9999, 10000), chunks=(5000, 5000), spec=spec
) # 200MB chunks
b = xp.flip(a, axis=0)
run_operation(tmp_path, executor, "flip", b)


@pytest.mark.slow
def test_reshape(tmp_path, spec, executor):
a = cubed.random.random(
Expand Down

0 comments on commit b1cf324

Please sign in to comment.