Initial commit

pyground/sugar/TestSugarRushSameSymbol.py

@@ -0,0 +1,51 @@
+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_same_symbol_clusters(self):
+        mock_grid = [
+            ['A','E','E','A','B','A','F'],
+            ['E','E','E','B','C','D','E'],
+            ['E','E','S','C','A','A','A'],
+            ['A','B','A','D','E','B','B'],
+            ['B','E','E','A','E','E','A'],
+            ['E','C','D','E','B','E','B'],
+            ['A','F','E','A','A','E','A'],
+        ]
+        game = SugarRush1000(balance=100000, bet=1, mock_grid=mock_grid)
+
+        # 1. 触发
+        import json
+        result = game.doSpin()
+        # print(json.dumps(result))
+        assert len(result['steps'][0]['symbol_links']) == 2
+        assert result['steps'][0]['symbol_links'][0]['symbol'] == 'E'
+        assert result['steps'][0]['symbol_links'][1]['symbol'] == 'E'
+        
+
+       
+        print("✅ 测试通过：同个符号多个cluster消除正常")
+
+
+
+if __name__ == "__main__":
+    unittest.main(argv=["first-arg-is-ignored"], exit=False)