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:47:31 +08:00
parent b54b46ca11
commit 15eaaa42b6
2 changed files with 128 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
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
from src.engine.fan_type import is_basic_win,is_cleared,calculate_terminal_fan,is_seven_pairs,is_full_request
from src.engine.meld import Meld
def test_is_basic_win():
@@ -100,4 +100,76 @@ def test_calculate_terminal_fan():
hand.add_tile(MahjongTile("条", 5))
hand.add_tile(MahjongTile("条", 5))
melds = []
assert calculate_terminal_fan(hand, melds) == 3, "测试失败：基本带幺九应为 3 番"
assert calculate_terminal_fan(hand, melds) == 3, "测试失败：基本带幺九应为 3 番"
def test_is_seven_pairs():
"""测试七对番型"""
hand = Hand()
# 示例1：符合七对
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))
assert is_seven_pairs(hand) == 2, "测试失败：符合七对，应为 2 番"
# 示例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))
assert is_seven_pairs(hand) == 0, "测试失败：不符合七对，应为 0 番"
print("所有七对测试通过！")
def test_is_full_request():
"""测试全求人番型"""
hand = Hand()
# 示例1：符合全求人
hand.add_tile(MahjongTile("筒", 5)) # 玩家手中只剩下 1 张牌
melds = [
Meld(MahjongTile("筒", 1), "碰"),
Meld(MahjongTile("筒", 2), "杠"),
Meld(MahjongTile("筒", 3), "碰"),
Meld(MahjongTile("筒", 4), "碰"),
]
winning_tile = MahjongTile("筒", 5) # 胡牌通过别人打出的牌
assert is_full_request(hand, melds, winning_tile) == 6, "测试失败：符合全求人，应为 6 番"
# 示例2：不符合全求人（玩家手上有多张牌）
hand = Hand()
hand.add_tile(MahjongTile("筒", 5))
hand.add_tile(MahjongTile("筒", 6))
assert is_full_request(hand, melds, winning_tile) == 0, "测试失败：不符合全求人，应为 0 番"
# 示例3：不符合全求人（没有碰或杠的明牌）
hand = Hand()
hand.add_tile(MahjongTile("筒", 5))
melds = []
assert is_full_request(hand, melds, winning_tile) == 0, "测试失败：不符合全求人，应为 0 番"
print("所有全求人测试通过！")