test(single-mode): add race menu test case

NateScarlet · Aug 19, 2022 · a2fe871 · a2fe871
1 parent d79182a
commit a2fe871
Show file tree

Hide file tree

Showing 4 changed files with 98 additions and 0 deletions.
diff --git a/auto_derby/data/ocr_labels.csv b/auto_derby/data/ocr_labels.csv
@@ -1107,3 +1107,5 @@ f81ffc1ffe0f1e001f003f00ff1fff3f3e7f007c007800780c7c3e3efe1ff80f,5
 00063f263f7f3f7f3f3b3f73ff733f1f3f0e3f3ef03ff027713e793ef07f3001,駿
 38003800fe01ff03ff03be7f9cff1c7f1c000c008e01ce01ce01c77fc7ff86ff,た
 380c3e1c0f380f380e383c3efc1ffc1f3e3c0ff807f007f007f01f7cfc1ff007,8
+00600c618ee38ee38ee38ee38ee38ee38ee38ee38ee38ee38ee387e307e303e1,川
+00080c1e0c7f0c7f0c77cc73cc7fcc7f0c700c601c6fff7fff7f3f7f017d0030,崎
diff --git a/auto_derby/single_mode/race/__snapshots__/race_test/test_find_by_race_menu_image_11.json b/auto_derby/single_mode/race/__snapshots__/race_test/test_find_by_race_menu_image_11.json
@@ -0,0 +1,87 @@
+[
+  [
+    {
+      "stadium": "川崎",
+      "name": "関東オークス",
+      "grade": 200,
+      "ground": 2,
+      "distance": 2100,
+      "permission": 2,
+      "month": 6,
+      "half": 1,
+      "entryCount": 14,
+      "track": 1,
+      "turn": 2,
+      "targetStatuses": [
+        2,
+        5
+      ],
+      "minFanCount": 1800,
+      "fanCounts": [
+        3500,
+        1400,
+        875,
+        525,
+        350,
+        315,
+        280,
+        245,
+        210,
+        175,
+        140,
+        105,
+        88,
+        70,
+        53,
+        35,
+        35,
+        35
+      ],
+      "gradePoints": [
+        80,
+        48,
+        32,
+        16,
+        16,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8,
+        8
+      ],
+      "shopCoins": [
+        100,
+        60,
+        60,
+        30,
+        30,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0,
+        0
+      ],
+      "characters": []
+    },
+    [
+      203,
+      655
+    ]
+  ]
+]
diff --git a/auto_derby/single_mode/race/race_test.py b/auto_derby/single_mode/race/race_test.py
@@ -304,6 +304,15 @@ def test_find_by_race_menu_image_10():
     _test.snapshot_match(res)
 
 
+def test_find_by_race_menu_image_11():
+    ctx = Context.new()
+    ctx.date = (2, 6, 1)
+    ctx.scenario = ctx.SCENARIO_CLIMAX
+    img, _ = _test.use_screenshot("single_mode/race_menu_11.png")
+    res = sorted(race.find_by_race_menu_image(ctx, img), key=lambda x: x[1][1])
+    _test.snapshot_match(res)
+
+
 def test_find_by_race_menu_image_issue112():
     ctx = Context.new()
     ctx.date = (1, 12, 1)

diff --git a/auto_derby/test_data/single_mode/race_menu_11.png b/auto_derby/test_data/single_mode/race_menu_11.png