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)
