def test_DATAIO_BATCHES_backup(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_batches(sample.img_data, sample.seg_data, "abc")
self.assertEqual(len(os.listdir(self.tmp_batches)), 2)
data_io.batch_cleanup()
def test_DATAIO_BATCHES_loading(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_batches(sample.img_data, sample.seg_data, "abc")
img = data_io.batch_load(pointer="abc", img=True)
self.assertTrue(np.array_equal(sample.img_data, img))
seg = data_io.batch_load(pointer="abc", img=False)
self.assertTrue(np.array_equal(sample.seg_data, seg))
data_io.batch_cleanup()
