class TestLocationFinderFilter(unittest.TestCase): def setUp(self): self.gui = GUI(GeistFakeBackend()) def test_filter(self): in_location = self.gui.wait_find_one(Location(10, 10, 100, 100)) parent_x, parent_y = 40, 50 child_x, child_y = 100, 17 finder = FinderInFinder(Location(child_x, child_y, 100, 100), Location(parent_x, parent_y, 500, 500)) result = self.gui.wait_find_one(finder) self.assertEqual(parent_x + child_x, result.x) self.assertEqual(parent_y + child_y, result.y)
def test_nontrivial_StopChangingFinder(self): backend = GeistFakeBackend(image=np.array([[[0], [0]], [[0], [0]]])) gui = GUI(backend) finder = StopChangingFinder(self.location) #backend.image = np.array([[[0],[0]],[[0],[0]]]) finder.period = 1 location_found = gui.wait_find_one(finder) self.assertTrue(np.array_equal(location_found.image, backend.image))
def test_nontrivial_StopChangingFinder(self): backend = GeistFakeBackend(image=np.array([[[0], [0]], [[0], [0]]])) gui = GUI(backend) finder = StopChangingFinder(self.location) # backend.image = np.array([[[0],[0]],[[0],[0]]]) finder.period = 1 location_found = gui.wait_find_one(finder) self.assertTrue(np.array_equal(location_found.image, backend.image))