def test_filter():
x = arange(2*3*4).reshape(2, 3, 4)
b = array(x)
# Test all generic filter functionality
generic.filter_suite(x, b)
def test_filter(sc):
x = arange(2*3*4).reshape(2, 3, 4)
b = array(x, sc, axis=0)
# Test all filter functionality when the base array is split after the first axis
generic.filter_suite(x, b)
# Split the BoltArraySpark after the second axis and rerun the tests
b = array(x, sc, axis=(0, 1))
generic.filter_suite(x, b)
# Split the BoltArraySpark after the third axis (scalar values) and rerun the tests
b = array(x, sc, axis=(0, 1, 2))
generic.filter_suite(x, b)
def test_filter(sc):
x = arange(2*3*4).reshape(2, 3, 4)
b = array(x, sc, axis=0)
# Test all filter functionality when the base array is split after the first axis
generic.filter_suite(x, b)
# Split the BoltArraySpark after the second axis and rerun the tests
b = array(x, sc, axis=(0, 1))
generic.filter_suite(x, b)
# Split the BoltArraySpark after the third axis (scalar values) and rerun the tests
b = array(x, sc, axis=(0, 1, 2))
generic.filter_suite(x, b)
# Test that the split is correct after the filter (should always be 1
# due to reindexing)
b = array(x, sc, axis=(0, 1))
r = b.filter(lambda x: True, axis=(1,))
assert r.shape == (b.shape[1], b.shape[0], b.shape[2])
assert r._split == 1
def test_filter(sc):
x = arange(2 * 3 * 4).reshape(2, 3, 4)
b = array(x, sc, axis=0)
# Test all filter functionality when the base array is split after the first axis
generic.filter_suite(x, b)
# Split the BoltArraySpark after the second axis and rerun the tests
b = array(x, sc, axis=(0, 1))
generic.filter_suite(x, b)
# Split the BoltArraySpark after the third axis (scalar values) and rerun the tests
b = array(x, sc, axis=(0, 1, 2))
generic.filter_suite(x, b)
# Test that the split is correct after the filter (should always be 1
# due to reindexing)
b = array(x, sc, axis=(0, 1))
r = b.filter(lambda x: True, axis=(1, ))
assert r.shape == (b.shape[1], b.shape[0], b.shape[2])
assert r._split == 1
