def test_redistribute(self):
g = memh5.BasicCont(distributed=True)
# Create an array from data
g.create_dataset("data",
shape=(10, 10),
distributed=True,
distributed_axis=0)
self.assertEqual(g["data"].distributed_axis, 0)
g.redistribute(1)
self.assertEqual(g["data"].distributed_axis, 1)
def test_access(self):
d = memh5.BasicCont()
self.assertTrue('history' in d._data.keys())
self.assertTrue('index_map' in d._data.keys())
self.assertRaises(KeyError, d.__getitem__, 'history')
self.assertRaises(KeyError, d.__getitem__, 'index_map')
self.assertRaises(ValueError, d.create_group, 'a')
self.assertRaises(ValueError, d.create_dataset, 'index_map/stuff',
data=np.arange(5))
# But make sure this works.
d.create_dataset('a', data=np.arange(5))
def test_access(self):
d = memh5.BasicCont()
self.assertTrue("history" in d._data)
self.assertTrue("index_map" in d._data)
self.assertRaises(KeyError, d.__getitem__, "history")
self.assertRaises(KeyError, d.__getitem__, "index_map")
self.assertRaises(ValueError, d.create_group, "a")
self.assertRaises(ValueError,
d.create_dataset,
"index_map/stuff",
data=np.arange(5))
# But make sure this works.
d.create_dataset("a", data=np.arange(5))
def process(self):
"""Produce an empty stream and pass on.
Returns
-------
cont : subclass of `memh5.BasicCont`
Empty data stream.
"""
if self.total_len == 0:
raise pipeline.PipelineStopIteration
self.log.debug("Producing test data '{}'...".format(self.tag))
cont = memh5.BasicCont()
if "tag" not in cont.attrs:
cont.attrs["tag"] = self.tag
self.total_len -= 1
return cont
def setUp(self):
d = memh5.BasicCont()
d.create_dataset("a", data=np.arange(5))
d.add_history("test", self.history_dict)
d.to_disk(self.fname)