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)