def test_columnwise_iteration_with_structured_array_and_mask():
input = numpy.arange(12).reshape((4,3))
m = LazyArray(input, shape=(4,3)) # 4 rows, 3 columns
mask = numpy.array([False, True, True])
cols = [col for col in m.by_column(mask=mask)]
assert_arrays_equal(cols[0], input[:,1])
assert_arrays_equal(cols[1], input[:,2])
def test_columnwise_iteration_with_function():
input = lambda i,j: 2*i + j
m = LazyArray(input, shape=(4,3))
cols = [col for col in m.by_column()]
assert_arrays_equal(cols[0], numpy.array([0, 2, 4, 6]))
assert_arrays_equal(cols[1], numpy.array([1, 3, 5, 7]))
assert_arrays_equal(cols[2], numpy.array([2, 4, 6, 8]))
def test_columnwise_iteration_with_structured_array_and_mask():
input = numpy.arange(12).reshape((4, 3))
m = LazyArray(input, shape=(4, 3)) # 4 rows, 3 columns
mask = numpy.array([False, True, True])
cols = [col for col in m.by_column(mask=mask)]
assert_arrays_equal(cols[0], input[:, 1])
assert_arrays_equal(cols[1], input[:, 2])
def test_columnwise_iteration_with_function():
input = lambda i, j: 2 * i + j
m = LazyArray(input, shape=(4, 3))
cols = [col for col in m.by_column()]
assert_arrays_equal(cols[0], numpy.array([0, 2, 4, 6]))
assert_arrays_equal(cols[1], numpy.array([1, 3, 5, 7]))
assert_arrays_equal(cols[2], numpy.array([2, 4, 6, 8]))
def test_columnwise_iteration_with_random_array_parallel_safe_no_mask():
random.mpi_rank=0; random.num_processes=2
input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
copy_input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
m = LazyArray(input, shape=(4,3))
cols = [col for col in m.by_column()]
assert_arrays_equal(cols[0], copy_input.next(4, mask_local=False))
assert_arrays_equal(cols[1], copy_input.next(4, mask_local=False))
assert_arrays_equal(cols[2], copy_input.next(4, mask_local=False))
def test_columnwise_iteration_with_random_array_parallel_safe_with_mask():
random.mpi_rank=0; random.num_processes=2
input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
copy_input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
m = LazyArray(input, shape=(4,3))
mask = numpy.array([False, False, True])
cols = [col for col in m.by_column(mask=mask)]
assert_equal(len(cols), 1)
assert_arrays_almost_equal(cols[0], copy_input.next(12, mask_local=False)[8:], 1e-15)
def test_columnwise_iteration_with_random_array_parallel_safe_no_mask():
random.mpi_rank = 0
random.num_processes = 2
input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
copy_input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
m = LazyArray(input, shape=(4, 3))
cols = [col for col in m.by_column()]
assert_arrays_equal(cols[0], copy_input.next(4, mask_local=False))
assert_arrays_equal(cols[1], copy_input.next(4, mask_local=False))
assert_arrays_equal(cols[2], copy_input.next(4, mask_local=False))
def test_columnwise_iteration_with_random_array_parallel_safe_with_mask():
random.mpi_rank = 0
random.num_processes = 2
input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
copy_input = random.RandomDistribution(rng=MockRNG(parallel_safe=True))
m = LazyArray(input, shape=(4, 3))
mask = numpy.array([False, False, True])
cols = [col for col in m.by_column(mask=mask)]
assert_equal(len(cols), 1)
assert_arrays_almost_equal(cols[0],
copy_input.next(12, mask_local=False)[8:],
1e-15)
def test_columnwise_iteration_with_structured_array():
input = numpy.arange(12).reshape((4,3))
m = LazyArray(input, shape=(4,3)) # 4 rows, 3 columns
cols = [col for col in m.by_column()]
assert_arrays_equal(cols[0], input[:,0])
assert_arrays_equal(cols[2], input[:,2])
def test_columnwise_iteration_with_flat_array():
m = LazyArray(5, shape=(4,3)) # 4 rows, 3 columns
cols = [col for col in m.by_column()]
assert_equal(cols, [5, 5, 5])
def test_columnwise_iteration_with_flat_array_and_mask():
m = LazyArray(5, shape=(4,3)) # 4 rows, 3 columns
mask = numpy.array([True, False, True])
cols = [col for col in m.by_column(mask=mask)]
assert_equal(cols, [5, 5])
def test_columnwise_iteration_with_flat_array_and_mask():
m = LazyArray(5, shape=(4, 3)) # 4 rows, 3 columns
mask = numpy.array([True, False, True])
cols = [col for col in m.by_column(mask=mask)]
assert_equal(cols, [5, 5])
def test_columnwise_iteration_with_structured_array():
input = numpy.arange(12).reshape((4, 3))
m = LazyArray(input, shape=(4, 3)) # 4 rows, 3 columns
cols = [col for col in m.by_column()]
assert_arrays_equal(cols[0], input[:, 0])
assert_arrays_equal(cols[2], input[:, 2])
def test_columnwise_iteration_with_flat_array():
m = LazyArray(5, shape=(4, 3)) # 4 rows, 3 columns
cols = [col for col in m.by_column()]
assert_equal(cols, [5, 5, 5])
