Initial commit

pyground/sugar/TestSugarRushGrid.py

@@ -0,0 +1,46 @@
+import unittest
+
+from SugarRush1000 import SugarRush1000
+
+def checkGrid(grid):
+    total_scotter_count = 0
+    for col in range(7):
+        scotter_count = 0
+        for row in range(7):
+            if grid[row][col] == "S":
+                scotter_count += 1
+                total_scotter_count += 1
+        if scotter_count > 1:
+            print(f"col{col} has {scotter_count} scotter")
+            return False
+    if total_scotter_count > 7:
+        print(f"total scotter count {total_scotter_count} > 7")
+        return False
+
+    return True
+
+class TestSugarRushLogic(unittest.TestCase):
+
+    def test_scotter_rule(self):
+        game = SugarRush1000(balance=100000, bet=1)
+
+        # 1. 触发
+        for _ in range(10000):
+            result = game.doSpin()
+            assert checkGrid(result['grid'])
+
+       
+        print("✅ 测试通过：生成scotter规则正常")
+
+    def test_free_spin_scotter_rule(self):
+        game = SugarRush1000(balance=100000, bet=1)
+        for _ in range(10000):
+            game.buy_free_spins('standard')
+            result = game.doSpin()
+            assert checkGrid(result['grid'])
+        print("✅ 测试通过：免费旋转scotter规则正常")
+
+
+
+if __name__ == "__main__":
+    unittest.main(argv=["first-arg-is-ignored"], exit=False)