fix(single-mode): recognize friend type partner

auto_derby/single_mode/training/partner.py

@@ -26,9 +26,10 @@ def _recognize_type_color(rp: mathtools.ResizeProxy, icon_img: Image) -> int:
         ((255, 151, 27), Partner.TYPE_POWER),
         ((255, 96, 156), Partner.TYPE_GUTS),
         ((3, 191, 126), Partner.TYPE_WISDOM),
+        ((255, 179, 22), Partner.TYPE_FRIEND),
     )
     for color, v in type_colors:
-        if imagetools.compare_color(icon_img.getpixel(type_pos), color) > 0.8:
+        if imagetools.compare_color(icon_img.getpixel(type_pos), color) > 0.9:
             return v
     return Partner.TYPE_OTHER
 
@@ -123,7 +124,8 @@ class Partner:
     TYPE_POWER: int = 3
     TYPE_GUTS: int = 4
     TYPE_WISDOM: int = 5
-    TYPE_OTHER: int = 6
+    TYPE_FRIEND: int = 6
+    TYPE_OTHER: int = 7
 
     def __init__(self):
         self.level = 0
@@ -163,6 +165,7 @@ def type_text(v: int) -> Text:
             Partner.TYPE_POWER: "pow",
             Partner.TYPE_GUTS: "gut",
             Partner.TYPE_WISDOM: "wis",
+            Partner.TYPE_FRIEND: "frd",
             Partner.TYPE_OTHER: "oth",
         }.get(v, f"unknown({v})")
 

auto_derby/single_mode/training/training_test.py

@@ -199,6 +199,28 @@ def test_update_by_training_scene_8():
         assert partner3.has_hint == False
 
 
+def test_update_by_training_scene_9():
+    with _test.screenshot("training_scene_9.png") as img:
+        training = Training.from_training_scene(img)
+    assert training.type == training.TYPE_SPEED
+    assert training.level == 2
+    assert training.speed == 19
+    assert training.stamina == 0
+    assert training.power == 10
+    assert training.guts == 0
+    assert training.wisdom == 0
+    assert training.skill == 3
+
+    assert len(training.partners) == 2, len(training.partners)
+    partner1, partner2 = training.partners
+    assert partner1.type == Partner.TYPE_FRIEND
+    assert partner1.level == 4
+    assert partner1.has_hint == False
+    assert partner2.type == Partner.TYPE_WISDOM
+    assert partner2.level == 5
+    assert partner2.has_hint == False
+
+
 def test_update_by_training_scene_issue9():
     img = (
         PIL.Image.open(_TEST_DATA_PATH / "training_scene_issue9.png")
@@ -323,7 +345,7 @@ def test_update_by_training_scene_issue156():
 
     assert len(training.partners) == 3, len(training.partners)
     partner1, partner2, partner3 = training.partners
-    assert partner1.type == Partner.TYPE_POWER
+    assert partner1.type == Partner.TYPE_FRIEND
     assert partner1.level == 2
     assert partner1.has_hint == False
     assert partner2.type == Partner.TYPE_SPEED