Пример #1
0
 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)
Пример #2
0
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)
Пример #3
0
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)
Пример #4
0
 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)