diff --git a/auto_derby/templates/team_race_choose_competitor.png b/auto_derby/templates/team_race_choose_competitor.png index 9198b971..6bf127d9 100644 Binary files a/auto_derby/templates/team_race_choose_competitor.png and b/auto_derby/templates/team_race_choose_competitor.png differ diff --git a/auto_derby/templates/team_race_choose_competitor.pos.png b/auto_derby/templates/team_race_choose_competitor.pos.png index 8d860c89..d458ccd6 100644 Binary files a/auto_derby/templates/team_race_choose_competitor.pos.png and b/auto_derby/templates/team_race_choose_competitor.pos.png differ diff --git a/auto_derby/templates/team_race_choose_competitor_test.py b/auto_derby/templates/team_race_choose_competitor_test.py new file mode 100644 index 00000000..2dd3d2e5 --- /dev/null +++ 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 new file mode 100644 index 00000000..769b8c51 Binary files /dev/null and b/auto_derby/templates/test_data/team_race_arena.png differ diff --git a/auto_derby/templates/test_data/team_race_arena_issue144.png b/auto_derby/templates/test_data/team_race_arena_issue144.png new file mode 100644 index 00000000..c682603f Binary files /dev/null and b/auto_derby/templates/test_data/team_race_arena_issue144.png differ