100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
"""测试任务3 - 使用poker_task1进行扑克牌EMD距离计算"""
|
||
import sys
|
||
import os
|
||
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
||
|
||
# 从poker_task3.task3模块导入函数
|
||
from poker_task3.task3 import parseTurnInput, calHandEquityHist, calPokerEmdTurn
|
||
from poker_task3.card import Card
|
||
|
||
|
||
def test_parse_turn_input():
|
||
"""测试TURN阶段输入解析"""
|
||
input_text = """
|
||
As Ks
|
||
7s 6s
|
||
8h 9d 9c Qh
|
||
"""
|
||
|
||
hand1, hand2, board = parseTurnInput(input_text)
|
||
|
||
assert len(hand1) == 2
|
||
assert len(hand2) == 2
|
||
assert len(board) == 4
|
||
|
||
assert str(hand1[0]) == "As"
|
||
assert str(hand1[1]) == "Ks"
|
||
assert str(hand2[0]) == "7s"
|
||
assert str(hand2[1]) == "6s"
|
||
|
||
print("✓ TURN输入解析测试通过")
|
||
|
||
|
||
def test_hand_equity_Hist():
|
||
"""测试手牌胜率分布计算"""
|
||
hole_cards = Card.parse_cards("As Ks")
|
||
board_cards = Card.parse_cards("8h 9d 9c Qh")
|
||
|
||
Hist = calHandEquityHist(hole_cards, board_cards)
|
||
|
||
assert len(Hist) == 30 # 胜率分布的区间数量（改进为30个区间）
|
||
assert all(isinstance(x, (int, float)) for x in Hist)
|
||
assert all(x >= 0 for x in Hist) # 所有值都应该是非负的
|
||
assert abs(sum(Hist) - 1.0) < 1e-10 # 分布总和应该为1（标准化后）
|
||
|
||
print("✓ 手牌胜率分布测试通过")
|
||
|
||
|
||
def test_poker_emd_calculation():
|
||
"""测试主要的EMD计算函数"""
|
||
input_text = """As Ks
|
||
7s 6s
|
||
8h 9d 9c Qh"""
|
||
|
||
distance = calPokerEmdTurn(input_text)
|
||
|
||
assert isinstance(distance, (int, float))
|
||
assert distance >= 0 # EMD总是非负的
|
||
|
||
print(f"✓ 扑克牌EMD计算测试通过 (距离: {distance:.3f})")
|
||
|
||
|
||
def test_different_hand_strengths():
|
||
"""测试不同强度手牌的EMD"""
|
||
# 强牌 vs 弱牌
|
||
input_text = """As Ks
|
||
2c 3d
|
||
8h 9d 9c Qh"""
|
||
|
||
distance = calPokerEmdTurn(input_text)
|
||
|
||
assert isinstance(distance, (int, float))
|
||
assert distance >= 0 # 应该有非负距离
|
||
|
||
print(f"✓ 不同手牌强度测试通过 (距离: {distance:.3f})")
|
||
|
||
|
||
def test_error_handling():
|
||
"""测试无效输入的错误处理"""
|
||
# 错误的行数
|
||
result = parseTurnInput("As Ks\n7s 6s")
|
||
assert result == (None, None, None), "应该返回(None, None, None)"
|
||
|
||
# 手牌中牌数错误
|
||
result = parseTurnInput("As Ks Qs\n7s 6s\n8h 9d 9c Qh")
|
||
assert result == (None, None, None), "应该返回(None, None, None)"
|
||
|
||
# 公共牌数错误
|
||
result = parseTurnInput("As Ks\n7s 6s\n8h 9d 9c")
|
||
assert result == (None, None, None), "应该返回(None, None, None)"
|
||
|
||
print("✓ 错误处理测试通过")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
test_parse_turn_input()
|
||
test_hand_equity_Hist()
|
||
test_poker_emd_calculation()
|
||
test_different_hand_strengths()
|
||
test_error_handling()
|
||
print("所有任务3测试通过! ✓") |