def test_equality(self):
x = 10
y = 10
z = 3
pixel_id = sitk.sitkUInt8
size = (x, y, z)
direction = (0, 1, 0, 1, 0, 0, 0, 0, 1)
image = sitk.Image([x, y, z], pixel_id)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection(direction)
dut1 = img.ImageProperties(image)
dut2 = img.ImageProperties(image)
self.assertTrue(dut1 == dut2)
self.assertFalse(dut1 != dut2)
image = sitk.Image([x, y, z], sitk.sitkInt8)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection(direction)
dut1 = img.ImageProperties(image)
self.assertTrue(dut1 == dut2)
self.assertFalse(dut1 != dut2)
image = sitk.Image([x, y, z], sitk.sitkVectorUInt8, 2)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection(direction)
dut1 = img.ImageProperties(image)
self.assertTrue(dut1 == dut2)
self.assertFalse(dut1 != dut2)
def test_non_equality(self):
x = 10
y = 10
z = 3
pixel_id = sitk.sitkUInt8
size = (x, y, z)
direction = (0, 1, 0, 1, 0, 0, 0, 0, 1)
image = sitk.Image([x, y, z], pixel_id)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection(direction)
dut1 = img.ImageProperties(image)
different_size = (x, y, 2)
# non-equal size
image = sitk.Image(different_size, pixel_id)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection(direction)
dut2 = img.ImageProperties(image)
self.assertFalse(dut1 == dut2)
self.assertTrue(dut1 != dut2)
# non-equal origin
image = sitk.Image(size, pixel_id)
image.SetOrigin(different_size)
image.SetSpacing(size)
image.SetDirection(direction)
dut2 = img.ImageProperties(image)
self.assertFalse(dut1 == dut2)
self.assertTrue(dut1 != dut2)
# non-equal spacing
different_size = (x, y, 2)
image = sitk.Image(size, pixel_id)
image.SetOrigin(size)
image.SetSpacing(different_size)
image.SetDirection(direction)
dut2 = img.ImageProperties(image)
self.assertFalse(dut1 == dut2)
self.assertTrue(dut1 != dut2)
# non-equal direction
different_size = (x, y, 2)
image = sitk.Image(size, pixel_id)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection((1, 0, 0, 0, 1, 0, 0, 0, 1))
dut2 = img.ImageProperties(image)
self.assertFalse(dut1 == dut2)
self.assertTrue(dut1 != dut2)
def test_is_two_dimensional(self):
x = 10
y = 10
image = sitk.Image([x, y], sitk.sitkUInt8)
dut = img.ImageProperties(image)
self.assertEqual(dut.is_two_dimensional(), True)
self.assertEqual(dut.is_three_dimensional(), False)
self.assertEqual(dut.is_vector_image(), False)
def test_is_vector_image(self):
x = 10
y = 10
number_of_components_per_pixel = 3
image = sitk.Image([x, y], sitk.sitkVectorUInt8,
number_of_components_per_pixel)
dut = img.ImageProperties(image)
self.assertEqual(dut.is_two_dimensional(), True)
self.assertEqual(dut.is_three_dimensional(), False)
self.assertEqual(dut.is_vector_image(), True)
def test_properties(self):
x = 10
y = 10
z = 3
pixel_id = sitk.sitkUInt8
size = (x, y, z)
direction = (0, 1, 0, 1, 0, 0, 0, 0, 1)
image = sitk.Image([x, y, z], pixel_id)
image.SetOrigin(size)
image.SetSpacing(size)
image.SetDirection(direction)
dut = img.ImageProperties(image)
self.assertEqual(dut.size, size)
self.assertEqual(dut.origin, size)
self.assertEqual(dut.spacing, size)
self.assertEqual(dut.direction, direction)
self.assertEqual(dut.dimensions, z)
self.assertEqual(dut.number_of_components_per_pixel, 1)
self.assertEqual(dut.pixel_id, pixel_id)