def test_dataio_slice_delegation(self): indata = np.arange(30) dset = DataIO(indata) self.assertTrue(np.all(dset[2:15] == indata[2:15])) indata = np.arange(50).reshape(5, 10) dset = DataIO(indata) self.assertTrue(np.all(dset[1:3, 5:8] == indata[1:3, 5:8]))
def test_dataio(self): """Test get_data_shape on DataIO of various shapes and maxshape.""" dio = DataIO(data=[1, 2]) res = get_data_shape(dio) self.assertTupleEqual(res, (2, )) dio = DataIO(data=[[1, 2], [3, 4], [5, 6]]) res = get_data_shape(dio) self.assertTupleEqual(res, (3, 2)) dio = DataIO(data=np.array([[1, 2], [3, 4], [5, 6]])) res = get_data_shape(dio) self.assertTupleEqual(res, (3, 2))
def test_set_dataio_data_already_set(self): """ Test that Data.set_dataio works as intended """ dataio = DataIO(data=np.arange(30).reshape(5, 2, 3)) data = np.arange(30).reshape(5, 2, 3) container = Data('wrapped_data', data) with self.assertRaisesWith(ValueError, "cannot overwrite 'data' on DataIO"): container.set_dataio(dataio)
def test_set_dataio(self): """ Test that Data.set_dataio works as intended """ dataio = DataIO() data = np.arange(30).reshape(5, 2, 3) container = Data('wrapped_data', data) container.set_dataio(dataio) self.assertIs(dataio.data, data) self.assertIs(dataio, container.data)
def test_deepcopy(self): obj = DataIO(data=[1., 2., 3.]) obj_copy = deepcopy(obj) self.assertNotEqual(id(obj), id(obj_copy)) self.assertNotEqual(id(obj.data), id(obj_copy.data))