wsy182/mjAi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1

Browse Source 
1
This commit is contained in:
wangsiyuan
2024-12-01 03:53:46 +08:00
parent 15eaaa42b6
commit 672c3ad43e
3 changed files with 111 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_calculate_fan.py
View File

@@ -270,3 +270,4 @@ def test_self_draw():
fan = calculate_fan(hand, melds, is_self_draw=True, is_cleared=False, conditions=conditions)
assert fan == 1, f"Expected 1 fan (self-draw), got {fan}"

72
tests/test_fan_type.py
View File

@@ -1,6 +1,6 @@
from src.engine.hand import Hand
from src.engine.mahjong_tile import MahjongTile
from src.engine.fan_type import is_basic_win,is_cleared,calculate_terminal_fan,is_seven_pairs,is_full_request
from src.engine.fan_type import is_basic_win,is_cleared,calculate_terminal_fan,is_seven_pairs,is_full_request,is_dragon_seven_pairs
from src.engine.meld import Meld
def test_is_basic_win():
@@ -173,3 +173,73 @@ def test_is_full_request():
print("所有全求人测试通过！")
def test_is_dragon_seven_pairs():
"""测试龙七对番型计算"""
# 示例1：符合龙七对
hand = Hand()
hand.add_tile(MahjongTile("筒", 1))
hand.add_tile(MahjongTile("筒", 1))
hand.add_tile(MahjongTile("筒", 2))
hand.add_tile(MahjongTile("筒", 2))
hand.add_tile(MahjongTile("筒", 3))
hand.add_tile(MahjongTile("筒", 3))
hand.add_tile(MahjongTile("筒", 4))
hand.add_tile(MahjongTile("筒", 4))
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 6))
hand.add_tile(MahjongTile("筒", 6))
hand.add_tile(MahjongTile("筒", 7)) # 四张7筒
hand.add_tile(MahjongTile("筒", 7))
hand.add_tile(MahjongTile("筒", 7))
hand.add_tile(MahjongTile("筒", 7))
melds = [] # 没有明牌
fan, root_adjustment = is_dragon_seven_pairs(hand, melds)
assert fan == 12 and root_adjustment == -1, "测试失败：符合龙七对，应为 12 番，并减少 1 根"
# 示例2：不符合龙七对（只有七对，没有四张）
hand = Hand()
hand.add_tile(MahjongTile("筒", 1))
hand.add_tile(MahjongTile("筒", 1))
hand.add_tile(MahjongTile("筒", 2))
hand.add_tile(MahjongTile("筒", 2))
hand.add_tile(MahjongTile("筒", 3))
hand.add_tile(MahjongTile("筒", 3))
hand.add_tile(MahjongTile("筒", 4))
hand.add_tile(MahjongTile("筒", 4))
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 6))
hand.add_tile(MahjongTile("筒", 6))
hand.add_tile(MahjongTile("筒", 7))
hand.add_tile(MahjongTile("筒", 7))
fan, root_adjustment = is_dragon_seven_pairs(hand, melds)
assert fan == 0 and root_adjustment == 0, "测试失败：不符合龙七对，应为 0 番，根数不变"
# 示例3：不符合龙七对（有明牌）
hand = Hand()
hand.add_tile(MahjongTile("筒", 1))
hand.add_tile(MahjongTile("筒", 1))
hand.add_tile(MahjongTile("筒", 2))
hand.add_tile(MahjongTile("筒", 2))
hand.add_tile(MahjongTile("筒", 3))
hand.add_tile(MahjongTile("筒", 3))
hand.add_tile(MahjongTile("筒", 4))
hand.add_tile(MahjongTile("筒", 4))
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 6))
hand.add_tile(MahjongTile("筒", 6))
hand.add_tile(MahjongTile("筒", 7))
hand.add_tile(MahjongTile("筒", 7))
melds = [Meld(MahjongTile("筒", 8), "碰")] # 有明牌
fan, root_adjustment = is_dragon_seven_pairs(hand, melds)
assert fan == 0 and root_adjustment == 0, "测试失败：有明牌，不符合龙七对，应为 0 番，根数不变"
print("所有龙七对测试通过！")