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))
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()
def test_DATAIO_BATCHES_cleanup(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") data_io.backup_batches(sample.img_data, sample.seg_data, "def") data_io.backup_batches(sample.img_data, None, pointer="ghi") self.assertEqual(len(os.listdir(self.tmp_batches)), 5) data_io.batch_cleanup(pointer="def") self.assertEqual(len(os.listdir(self.tmp_batches)), 3) data_io.batch_cleanup() self.assertEqual(len(os.listdir(self.tmp_batches)), 0)