47 lines
1.3 KiB
Python
47 lines
1.3 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_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)
|