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)