more correctness in testy ways (ab86aa1f) · Commits · aflab / astrogeology / Autocnet

autocnet/matcher/tests/test_naive_template.py

+11 −10

Original line number	Original line	Diff line number	Diff line
	@@ -77,10 +77,11 @@ class TestNaiveTemplate(unittest.TestCase):

	def test_t_shape(self):		def test_t_shape(self):
	result_x, result_y, result_strength, _ = naive_template.pattern_match(self._t_shape,		result_x, result_y, result_strength, _ = naive_template.pattern_match(self._t_shape,
	self._test_image, upsampling=1)		self._test_image,
			upsampling=4)
	# Test offsets		# Test offsets
	self.assertEqual(result_x, 3)		self.assertEqual(result_x, -3)
	self.assertEqual(result_y, 3)		self.assertEqual(result_y, -3)
	# Test Correlation Strength: At least 0.8		# Test Correlation Strength: At least 0.8
	self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)		self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)

	@@ -88,17 +89,17 @@ class TestNaiveTemplate(unittest.TestCase):
	result_x, result_y, result_strength, _ = naive_template.pattern_match(self._rect_shape,		result_x, result_y, result_strength, _ = naive_template.pattern_match(self._rect_shape,
	self._test_image, upsampling=1)		self._test_image, upsampling=1)
	# Test offsets		# Test offsets
	self.assertEqual(result_x, -3)		self.assertEqual(result_x, 3)
	self.assertEqual(result_y, -4)		self.assertEqual(result_y, 4)
	# Test Correlation Strength: At least 0.8		# Test Correlation Strength: At least 0.8
	self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)		self.assertGreaterEqual(result_strength, 0.99, "Returned Correlation Strength of %d" % result_strength)

	def test_square_shape(self):		def test_square_shape(self):
	result_x, result_y, result_strength, _ = naive_template.pattern_match(self._square_shape,		result_x, result_y, result_strength, _ = naive_template.pattern_match(self._square_shape,
	self._test_image, upsampling=1)		self._test_image, upsampling=1)
	# Test offsets		# Test offsets
	self.assertEqual(result_x, 2)		self.assertEqual(result_x, -2)
	self.assertEqual(result_y, -4)		self.assertEqual(result_y, 4)
	# Test Correlation Strength: At least 0.8		# Test Correlation Strength: At least 0.8
	self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)		self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)

	@@ -106,8 +107,8 @@ class TestNaiveTemplate(unittest.TestCase):
	result_x, result_y, result_strength, _ = naive_template.pattern_match(self._vertical_line,		result_x, result_y, result_strength, _ = naive_template.pattern_match(self._vertical_line,
	self._test_image, upsampling=1)		self._test_image, upsampling=1)
	# Test offsets		# Test offsets
	self.assertEqual(result_x, -3)		self.assertEqual(result_x, 3)
	self.assertEqual(result_y, 5)		self.assertEqual(result_y, -5)
	# Test Correlation Strength: At least 0.8		# Test Correlation Strength: At least 0.8
	self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)		self.assertGreaterEqual(result_strength, 0.8, "Returned Correlation Strength of %d" % result_strength)