def test_iterate_over_all(self):
data = iter(
test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={}))
with self.assertRaises(StopIteration):
while True:
next(data)
self.assertEqual(len(data), 49)
def test_len(self):
dataset = iter(
test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={}))
self.assertEqual(len(dataset), 0)
next(dataset)
self.assertEqual(len(dataset), 1)
next(dataset)
next(dataset)
self.assertEqual(len(dataset), 3)
def test_summarize_correct_usage(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
self.assertIsInstance(data.summarize("Modality", print_summary=False),
test.gem.LabelCounter)
def test_classes_wrong_label_type(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
with self.assertRaises(TypeError):
data.classes(["Modality"])
def test_classes_correct_usage(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
self.assertIsInstance(data.classes("Modality"), list)
def test_can_be_parallelized(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
self.assertEqual(data.can_be_parallelized(), False)
def test_subset_wrong_constraint_type(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
with self.assertRaises(TypeError):
subset = data.subset(("Modality", 1))
def test_init_wrong_label_counter_type(self):
with self.assertRaises(TypeError):
test.PickledDicomoDataFolder(
dicom_directory, ["CT"],
constraints={},
label_counter_type=TestPickledDicomoDataSet)
def test_iter(self):
dataset = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
dataset = iter(dataset)
self.assertIsInstance(dataset, test.PickledDicomoDataFolder)
def test_next(self):
dataset = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
data = next(iter(dataset))
self.assertIsInstance(data, list)
def test_init_wrong_constraints_type(self):
with self.assertRaises(TypeError):
dataset = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints=[])
def test_init_file_has_wrong_type(self):
dataset = test.PickledDicomoDataFolder(raw_dicom_directory, ["CT"],
constraints={})
with self.assertRaises(test.gem.pickle.UnpicklingError):
next(iter(dataset))
def test_init_invalid_directory_path(self):
with self.assertRaises(NotADirectoryError):
test.PickledDicomoDataFolder(os.path.join(dicom_directory, "asd"),
["CT"],
constraints={})
def test_summarize_test_CT_constraint(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
summary_1 = data.summarize("Modality", print_summary=False)
summary_2 = data.summarize("BodyPartExamined", print_summary=False)
self.assertNotEqual(str(summary_1), str(summary_2))
def test_subset_correct_usage(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
subset = data.subset({"Modality": "asd"})
with self.assertRaises(StopIteration):
next(iter(subset))
def test_summarize_wrong_summary_type(self):
data = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
with self.assertRaises(TypeError):
data.summarize("Modality", print_summary=None)
def test_init_correct_usage(self):
dataset = test.PickledDicomoDataFolder(dicom_directory, ["CT"],
constraints={})
self.assertIsInstance(dataset, test.PickledDicomoDataFolder)