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)