fix(team-race): update template for adb device

fix #144
NateScarlet · Aug 16, 2021 · 2f265db · 2f265db
1 parent 4ff2808
commit 2f265db
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 0 deletions.
diff --git a/auto_derby/templates/team_race_choose_competitor.png b/auto_derby/templates/team_race_choose_competitor.png
diff --git a/auto_derby/templates/team_race_choose_competitor.pos.png b/auto_derby/templates/team_race_choose_competitor.pos.png
diff --git a/auto_derby/templates/team_race_choose_competitor_test.py b/auto_derby/templates/team_race_choose_competitor_test.py
@@ -0,0 +1,22 @@
+from auto_derby import mathtools
+from . import _test
+from .. import template, templates
+
+
+_EXPECTED_POS = (240, 154)
+
+
+def test_match():
+    img, rp = _test.use_screenshot("team_race_arena.png")
+    res = tuple(template.match(img, templates.TEAM_RACE_CHOOSE_COMPETITOR))
+    assert len(res) == 1
+    (match1,) = res
+    assert match1[1] == rp.vector2(_EXPECTED_POS, 540)
+
+
+def test_match_issue135():
+    img, rp = _test.use_screenshot("team_race_arena_issue144.png")
+    res = tuple(template.match(img, templates.TEAM_RACE_CHOOSE_COMPETITOR))
+    assert len(res) == 1
+    (match1,) = res
+    assert mathtools.distance(match1[1], rp.vector2(_EXPECTED_POS, 540)) < 8
diff --git a/auto_derby/templates/test_data/team_race_arena.png b/auto_derby/templates/test_data/team_race_arena.png
diff --git a/auto_derby/templates/test_data/team_race_arena_issue144.png b/auto_derby/templates/test_data/team_race_arena_issue144.png