def func(shape, ndim, ordering, value): table = PackedTable(shape, ndim=ndim, ordering=ordering, value=value) table_local = table.scatter() table_global = table_local.gather() assert_equal(table_global.shape, table.shape) assert_equal(table_global.ndim, table.ndim) assert_equal(table_global._index, table._index) assert_equal(table_global.removed, table.removed) assert_equal(table_global.value, table.value)
def test_scatter(): np.random.seed(0) n = 4 x = np.random.random(n) layout = PackedTable(n, x=x) s = split(n, size, rank) scattered = layout.scatter() assert_same(scattered.x, x[s]) assert_same(scattered.all.x, x)