def test_array_partition_views():
flat = np.arange(14).astype('float64')
partition = make_dictlist()
views = array_partition_views(flat, partition)
assert np.allclose(views['bar'], np.arange(4).reshape((2, 2)))
assert np.allclose(views['fank']['fenk'][0], 4.)
assert np.allclose(views['fank']['funk'], np.array([6, 7]).reshape((2, 1)))
def test_array_partition_views():
flat = np.arange(14).astype("float64")
partition = make_dictlist()
views = array_partition_views(flat, partition)
assert np.allclose(views["bar"], np.arange(4).reshape((2, 2)))
assert np.allclose(views["fank"]["fenk"][0], 4.0)
assert np.allclose(views["fank"]["funk"], np.array([6, 7]).reshape((2, 1)))
def test_array_partition_views():
flat = np.arange(1024).astype('float64')
partition = {
'bla': 2,
'blo': [(2, 3), 5],
'blu': {'foo': (2, 4),
'far': [1, 2]},
'blubb': (2, 3),
}
views = array_partition_views(flat, partition)
assert views['bla'].shape == (2,)
assert views['blo'][0].shape == (2, 3)
assert views['blo'][1].shape == (5,)
assert views['blu']['foo'].shape == (2, 4)
assert views['blu']['far'][0].shape == (1,)
assert views['blu']['far'][1].shape == (2,)
assert views['blubb'].shape == (2, 3)
def test_array_partition_views():
flat = np.arange(1024).astype('float64')
partition = {
'bla': 2,
'blo': [(2, 3), 5],
'blu': {
'foo': (2, 4),
'far': [1, 2]
},
'blubb': (2, 3),
}
views = array_partition_views(flat, partition)
assert views['bla'].shape == (2, )
assert views['blo'][0].shape == (2, 3)
assert views['blo'][1].shape == (5, )
assert views['blu']['foo'].shape == (2, 4)
assert views['blu']['far'][0].shape == (1, )
assert views['blu']['far'][1].shape == (2, )
assert views['blubb'].shape == (2, 3)
