def test_DATAIO_BATCHES_sampleStorage(self):
data_io = Data_IO(self.io_interface, input_path="", output_path="",
batch_path=self.tmp_batches, delete_batchDir=False)
sample = data_io.sample_loader("TEST.sample_0", backup=False,
load_seg=True, load_pred=False)
data_io.backup_sample(sample)
self.assertEqual(len(os.listdir(self.tmp_batches)), 1)
data_io.batch_cleanup()
def test_DATAIO_BATCHES_sampleLoading(self):
data_io = Data_IO(self.io_interface, input_path="", output_path="",
batch_path=self.tmp_batches, delete_batchDir=False)
sample = data_io.sample_loader("TEST.sample_0", backup=False,
load_seg=True, load_pred=False)
data_io.backup_sample(sample)
sample_new = data_io.load_sample_pickle(sample.index)
data_io.batch_cleanup()
self.assertTrue(np.array_equal(sample_new.img_data, sample.img_data))
self.assertTrue(np.array_equal(sample_new.seg_data, sample.seg_data))
