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