def test_no_match(self):
blank = np.zeros((100, 100))
template = np.array([[0, 1], [1, 0]])
expected = []
result = is_similar(template, blank)
self.assertFalse(result)
def test_similar_size(self):
template = np.array([[0, 0, 0, 0, 0, 0],
[0, 0, 0, 1, 0, 0],
[0, 0, 1, 0, 0, 0],
[0, 1, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0]])
result = is_similar(template, self.image, size_tolerance=0.2)
self.assertTrue(result)
def test_match(self):
image = np.array([[0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 1, 0],
[0, 0, 0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 1, 0, 0, 0, 0, 0],
[0, 1, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0]])
template = np.array([[0, 1], [1, 0]])
result = is_similar(template, image)
self.assertEqual(result, "matches exist- size different")
def test_colour_size_different(self):
template = np.array([[[255,0,0],[240,10,10]]], dtype=np.int32)
result = is_similar(template, self.image3)
self.assertEqual(result, "matches exist- size different")
def test_colour(self):
result = is_similar(self.image3, self.image3)
self.assertTrue(result)