Skip to content

Commit

Permalink
Closes Bears-R-Us#3912: failing unit test test_is_locally_sorted_mult…
Browse files Browse the repository at this point in the history
…idim under gasnet nl==2
  • Loading branch information
ajpotts committed Nov 21, 2024
1 parent b456d3e commit 7b63b6d
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tests/pdarrayclass_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ def test_reshape(self):
assert r.shape == (2, 2)
assert isinstance(r, ak.pdarray)


def test_shape(self):
a = ak.arange(4)
np_a = np.arange(4)
Expand All @@ -49,7 +50,7 @@ def test_flatten(self, size):

@pytest.mark.skip_if_max_rank_less_than(3)
@pytest.mark.parametrize("size", pytest.prob_size)
def test_flatten(self, size):
def test_flatten_multidim(self, size):
size = size - (size % 4)
a = ak.arange(size)
b = a.reshape((2, 2, size / 4))
Expand Down Expand Up @@ -114,12 +115,13 @@ def test_is_locally_sorted_multi_locale(self, size):
assert not is_sorted(a)

@pytest.mark.skip_if_max_rank_less_than(3)
@pytest.mark.skip_if_nl_greater_than(2)
@pytest.mark.parametrize("dtype", DTYPES)
@pytest.mark.parametrize("axis", [None, 0, 1, (0, 2), (0, 1, 2)])
def test_is_locally_sorted_multidim(self, dtype, axis):
from arkouda.pdarrayclass import is_locally_sorted

a = ak.array(ak.randint(0, 100, (5, 7, 4), dtype=ak.int64, seed=SEED))
a = ak.array(ak.randint(0, 100, (10, 10, 10), dtype=ak.int64, seed=SEED))
sorted = is_locally_sorted(a, axis=axis)
if isinstance(sorted, np.bool_):
assert not sorted
Expand Down

0 comments on commit 7b63b6d

Please sign in to comment.