52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
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)
|