def test_takes_sequences(self):
img = [[1, 2, 3, 4, 5]]
template = [[1, 0]]
ght(img, template)
img = (1, 2, 3, 4, 5)
template = (1, 0)
ght(img, template)
def test_dimensions(self):
# 1D
img = scipy.rand(10)
template = scipy.random.randint(0, 2, (3))
result = ght(img, template)
self.assertEqual(result.ndim, 1, "Computing ght with one-dimensional input data failed.")
# 2D
img = scipy.rand(10, 11)
template = scipy.random.randint(0, 2, (3, 4))
result = ght(img, template)
self.assertEqual(result.ndim, 2, "Computing ght with two-dimensional input data failed.")
# 3D
img = scipy.rand(10, 11, 12)
template = scipy.random.randint(0, 2, (3, 4, 5))
result = ght(img, template)
self.assertEqual(result.ndim, 3, "Computing ght with three-dimensional input data failed.")
# 4D
img = scipy.rand(10, 11, 12, 13)
template = scipy.random.randint(0, 2, (3, 4, 5, 6))
result = ght(img, template)
self.assertEqual(result.ndim, 4, "Computing ght with four-dimensional input data failed.")
# 5D
img = scipy.rand(3, 4, 3, 4, 3)
template = scipy.random.randint(0, 2, (2, 2, 2, 2, 2))
result = ght(img, template)
self.assertEqual(result.ndim, 5, "Computing ght with five-dimensional input data failed.")
def test_takes_sequences(self):
img = [[1,2,3,4,5]]
template = [[1,0]]
ght(img, template)
img = ((1,2,3,4,5))
template = ((1,0))
ght(img, template)
def test_dimensions(self):
# 1D
img = scipy.rand(10)
template = scipy.random.randint(0, 2, (3))
result = ght(img, template)
self.assertEqual(result.ndim, 1, 'Computing ght with one-dimensional input data failed.')
# 2D
img = scipy.rand(10, 11)
template = scipy.random.randint(0, 2, (3, 4))
result = ght(img, template)
self.assertEqual(result.ndim, 2, 'Computing ght with two-dimensional input data failed.')
# 3D
img = scipy.rand(10, 11, 12)
template = scipy.random.randint(0, 2, (3, 4, 5))
result = ght(img, template)
self.assertEqual(result.ndim, 3, 'Computing ght with three-dimensional input data failed.')
# 4D
img = scipy.rand(10, 11, 12, 13)
template = scipy.random.randint(0, 2, (3, 4, 5, 6))
result = ght(img, template)
self.assertEqual(result.ndim, 4, 'Computing ght with four-dimensional input data failed.')
# 5D
img = scipy.rand(3, 4, 3, 4, 3)
template = scipy.random.randint(0, 2, (2, 2, 2, 2, 2))
result = ght(img, template)
self.assertEqual(result.ndim, 5, 'Computing ght with five-dimensional input data failed.')
def test_float_img(self):
# prepare
img = [[2.0, 3.0, 0, 0], [1.0, 2.0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
img = scipy.asarray(img)
template = scipy.asarray([[True, True], [True, False]])
result_array = scipy.asarray([[6.0, 5.0, 0, 0], [3.0, 2.0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]).astype(img.dtype)
result_dtype = img.dtype
# run
result = ght(img, template)
# test
self.assertTrue(
scipy.all(result == result_array), "Returned hough transformation differs from the expected values."
)
self.assertTrue(
result.dtype == result_dtype, "Returned hough transformation is not of the expected scipy.dtype"
)
def test_odd_template(self):
# prepare
img = [[1, 1, 1, 0, 0], [1, 1, 1, 0, 0], [1, 1, 1, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]]
img = scipy.asarray(img).astype(scipy.bool_)
template = scipy.asarray([[True, True, True], [True, True, True], [True, True, True]])
result_array = scipy.asarray(
[[4, 6, 4, 2, 0], [6, 9, 6, 3, 0], [4, 6, 4, 2, 0], [2, 3, 2, 1, 0], [0, 0, 0, 0, 0]]
).astype(scipy.int32)
result_dtype = scipy.int32
# run
result = ght(img, template)
# test
self.assertTrue(
scipy.all(result == result_array), "Returned hough transformation differs from the expected values."
)
self.assertTrue(
result.dtype == result_dtype, "Returned hough transformation is not of the expected scipy.dtype"
)
def test_int_img(self):
# prepare
img = [[2, 1, 0, 0],
[1, 1, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]]
img = scipy.asarray(img)
template = scipy.asarray([[True, True],
[True, False]])
result_array = scipy.asarray([[4, 2, 0, 0],
[2, 1, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]]).astype(img.dtype)
result_dtype = img.dtype
# run
result = ght(img, template)
#test
self.assertTrue(scipy.all(result == result_array), 'Returned hough transformation differs from the expected values.')
self.assertTrue(result.dtype == result_dtype, 'Returned hough transformation is not of the expected scipy.dtype')
def test_odd_template(self):
# prepare
img = [[1, 1, 1, 0, 0],
[1, 1, 1, 0, 0],
[1, 1, 1, 0, 0],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0]]
img = scipy.asarray(img).astype(scipy.bool_)
template = scipy.asarray([[True, True, True],
[True, True, True],
[True, True, True]])
result_array = scipy.asarray([[4, 6, 4, 2, 0],
[6, 9, 6, 3, 0],
[4, 6, 4, 2, 0],
[2, 3, 2, 1, 0],
[0, 0, 0, 0, 0]]).astype(scipy.int32)
result_dtype = scipy.int32
# run
result = ght(img, template)
#test
self.assertTrue(scipy.all(result == result_array), 'Returned hough transformation differs from the expected values.')
self.assertTrue(result.dtype == result_dtype, 'Returned hough transformation is not of the expected scipy.dtype')
