Formatting, improve GG check

replays/sc2readerplugins/statistics.py

@@ -27,7 +27,8 @@ def loserDoesGG(replay):
     loser_sids = [p.sid for p in replay.players if p.result == "Loss"]
     loser_messages = [m for m in replay.messages if m.pid in loser_sids]
     return any(
-        levenshtein(m.text.lower(), g) < 2 and m.text.lower() != "bg"
+        set((m.text.lower())) - set("g") == set()
+        or (levenshtein(m.text.lower(), g) < 2 and m.text.lower() != "bg")
         for g in GGS
         for m in loser_messages
     )

tests/integration/test_obs_tools.py

@@ -26,5 +26,3 @@ def test_sc2client_get_opponent():
     print(f"Is barcode: {barcode}")
 
     assert opponent is not None
-
-

tests/unit/test_sc2client.py

@@ -1,6 +1,4 @@
-from config import config
-from obs_tools.sc2client import SC2Client
-from obs_tools.types import GameInfo, Screen, UIInfo
+from obs_tools.types import Screen, UIInfo
 
 
 def test_uiinfo_equality_loading():
@@ -11,7 +9,21 @@ def test_uiinfo_equality_loading():
 
 
 def test_uiinfo_equality_menus():
-    ui1 = UIInfo(activeScreens=[Screen.background, Screen.foreground, Screen.navigation, Screen.home ])
-    ui2 = UIInfo(activeScreens=[Screen.home, Screen.background, Screen.foreground, Screen.navigation])
+    ui1 = UIInfo(
+        activeScreens=[
+            Screen.background,
+            Screen.foreground,
+            Screen.navigation,
+            Screen.home,
+        ]
+    )
+    ui2 = UIInfo(
+        activeScreens=[
+            Screen.home,
+            Screen.background,
+            Screen.foreground,
+            Screen.navigation,
+        ]
+    )
 
-    assert ui1 == ui2
+    assert ui1 == ui2