def test_transform_data_handles_empty_datasets(self):
Xtr_orig = np.arange(10).reshape(10, 1, 1)*10
Ytr_orig = -np.arange(10)
some_dataset = ImageDataset(Xtr_orig, Ytr_orig, None, None)
Xtr, Ytr = some_dataset.transform_data(np.array([]), np.array([]))
assert Xtr.shape == (0,1,1)
def test_transform_data_restitutes_shape(self):
Xtr_orig = np.arange(10).reshape(10, 1, 1)*10
Ytr_orig = -np.arange(10)
some_dataset = ImageDataset(Xtr_orig, Ytr_orig, None, None)
Xtr, Ytr = some_dataset.transform_data(some_dataset.Xtr, some_dataset.Ytr)
assert Xtr.shape == Xtr_orig.shape