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()