def test_draw_elp(self): """Test ellipse/circle drawing""" self.set_sample_images_io(self.image2, 'result_elp.dcm') self.in_args.extend(['-elp', '63.5', '63.5', '20', '20', '1023', '1.0', '1']) self.test_args = dcm_transform.parse_arguments(self.in_args) file_count, dataset = self.instanciate_sut_transform(self.test_args) self.assertEqual(file_count, 1) self.assertIsNotNone(dataset.pixel_array)
def test_draw_roi(self): """Test ROI drawing""" self.set_sample_images_io(self.image2, 'result_roi.dcm') self.in_args.extend(['-roi', '62', '62', '4', '1023']) self.test_args = dcm_transform.parse_arguments(self.in_args) file_count, dataset = self.instanciate_sut_transform(self.test_args) self.assertEqual(file_count, 1) self.assertIsNotNone(dataset.pixel_array)
def test_anon1(self): """Test anonymization (with dpt) """ str_anon = 'test_anon1' self.in_args.extend(['-an', str_anon, '-dpt']) self.test_args = dcm_transform.parse_arguments(self.in_args) # needed because dataset.walk(PN_Callback) does not give an optional data param: dcm_transform.ARGS = self.test_args dcm_transform.check_if_anonymize_or_cleanup_needed(self.dataset, self.test_args, True, True)
def test_draw_crosshair(self): """Test crosshairdrawing""" self.set_sample_images_io(self.image2, 'result_crosshair.dcm') self.in_args.extend(['-crosshair', \ '30', '82', '5', '1', '1024', '1.0', '60', '92', '10', '2', '1024', '1.0']) self.test_args = dcm_transform.parse_arguments(self.in_args) file_count, dataset = self.instanciate_sut_transform(self.test_args) self.assertEqual(file_count, 1) self.assertIsNotNone(dataset.pixel_array)
def test_draw_frect(self): """Test filled rectangle drawing""" self.set_sample_images_io(self.image2, 'result_frect.dcm') self.in_args.extend(['-frect',\ '50', '10', '20', '40', '500', '0.5', \ '80', '80', '30', '30', '500', '.3']) self.test_args = dcm_transform.parse_arguments(self.in_args) file_count, dataset = self.instanciate_sut_transform(self.test_args) self.assertEqual(file_count, 1) self.assertIsNotNone(dataset.pixel_array)
def test_draw_pixel(self): """Test pixel drawing""" self.set_sample_images_io(self.image2, 'result_pixel.dcm') self.in_args.extend(['-pixel', \ '35', '12', '1023', '1.0', \ '70', '92', '1023', '0.5']) self.test_args = dcm_transform.parse_arguments(self.in_args) file_count, dataset = self.instanciate_sut_transform(self.test_args) self.assertEqual(file_count, 1) self.assertIsNotNone(dataset.pixel_array)
def test_patient_tags(self): """Test common patient tags settings""" self.in_args.extend(['-pid', '1234', '-pname', 'doe^john', '-dob', '19420402']) self.test_args = dcm_transform.parse_arguments(self.in_args) file_count, dataset = self.instanciate_sut_transform(self.test_args) self.assertEqual(file_count, 1) self.assertEqual(dataset.PatientName, 'doe^john') self.assertEqual(dataset.PatientID, '1234') self.assertEqual(dataset.PatientBirthDate, '19420402')
def setUp(self): self.dataset = dicom.read_file(self.input_ds_path) self.in_args = [self.input_ds_path, self.output_ds_path] self.test_args = dcm_transform.parse_arguments(self.in_args)