def test_cards(self):
"""
"""
handcards1 = [roomai.common.PokerCard(0, 0), roomai.common.PokerCard(0, 1)]
handcards2 = [roomai.common.PokerCard(3, 1), roomai.common.PokerCard(3, 2)]
keepcards = [roomai.common.PokerCard(0, 2), roomai.common.PokerCard(0, 3), roomai.common.PokerCard(2, 0), roomai.common.PokerCard(2, 1), roomai.common.PokerCard(3, 3)]
pattern = TexasHoldemEnv.__cards2pattern_cards__(handcards1, keepcards)[0]
cards = TexasHoldemEnv.__cards2pattern_cards__(handcards1, keepcards)[1]
self.assertEqual(pattern[0],'4_1')
self.assertEqual(pattern[1],False)
self.assertEqual(pattern[2],True)
self.assertEqual(pattern[3],False)
self.assertEqual(pattern[4][0], 4)
self.assertEqual(pattern[4][1], 1)
self.assertEqual(cards[0].point_rank, 0)
self.assertEqual(cards[1].point_rank, 0)
self.assertEqual(cards[2].point_rank, 0)
self.assertEqual(cards[3].point_rank, 0)
self.assertEqual(cards[4].point_rank, 3)
self.assertEqual(cards[0].suit_rank, 0)
self.assertEqual(cards[1].suit_rank, 1)
self.assertEqual(cards[2].suit_rank, 2)
self.assertEqual(cards[3].suit_rank, 3)
self.assertEqual(cards[4].suit_rank, 3)
pattern1 = TexasHoldemEnv.__cards2pattern_cards__(handcards1, keepcards)
pattern2 = TexasHoldemEnv.__cards2pattern_cards__(handcards2, keepcards)
diff = TexasHoldemEnv.__compare_handcards__(handcards1, handcards2, keepcards)
self.assertTrue(diff > 0)
def test_compare(self):
"""
"""
h1 = [roomai.common.PokerCard(7, 0), roomai.common.PokerCard(7, 1)]
h2 = [roomai.common.PokerCard(2, 0), roomai.common.PokerCard(2, 1)]
h3 = [roomai.common.PokerCard(2, 2), roomai.common.PokerCard(2, 3)]
k = [roomai.common.PokerCard(3, 1), roomai.common.PokerCard(4, 2), roomai.common.PokerCard(5, 3), roomai.common.PokerCard(6, 0), roomai.common.PokerCard(7, 2)]
p1 = TexasHoldemEnv.__cards2pattern_cards__(h1, k)
p2 = TexasHoldemEnv.__cards2pattern_cards__(h2, k)
p3 = TexasHoldemEnv.__cards2pattern_cards__(h3, k)
def test_is_action_valid(self):
"""
"""
env = TexasHoldemEnv()
env.init()
print (TexasHoldemAction.AllIn)
action = TexasHoldemAction("Allin_1000")
print (action.key)
self.assertTrue(env.is_action_valid(action, env.public_state, env.person_states[env.public_state.turn]))
def test_pattern(self):
"""
"""
handcards1 =[roomai.common.PokerCard(0, 0), roomai.common.PokerCard(1, 1)]
keepcards =[roomai.common.PokerCard(2, 2), roomai.common.PokerCard(3, 3), roomai.common.PokerCard(4, 0), roomai.common.PokerCard(5, 1), roomai.common.PokerCard(6, 2)]
pattern = TexasHoldemEnv.__cards2pattern_cards__(handcards1, keepcards)
def test_available_actions(self):
"""
"""
env = TexasHoldemEnv()
env.init()
actions = TexasHoldemEnv.available_actions(
env.public_state, env.person_states[env.public_state.turn])
self.assertTrue("Allin_1000" in actions)
env.public_state.raise_account = 200
actions = TexasHoldemEnv.available_actions(
env.public_state, env.person_states[env.public_state.turn])
self.assertTrue("Call_10" in actions)
self.assertTrue("Raise_210" in actions)
self.assertTrue("Raise_410" in actions)
self.assertTrue("Raise_410" in actions)
self.assertTrue("Raise_810" in actions)
self.assertTrue("Allin_1000" in actions)
for key in actions:
act = actions[key]
self.assertTrue(
TexasHoldemEnv.is_action_valid(
act, env.public_state,
env.person_states[env.public_state.turn]))
def test_cards2(self):
"""
"""
h1 = [roomai.common.PokerCard(7, 0), roomai.common.PokerCard(7, 1)]
keep = [roomai.common.PokerCard(3, 1), roomai.common.PokerCard(4, 2), roomai.common.PokerCard(5, 3), roomai.common.PokerCard(6, 0), roomai.common.PokerCard(7, 2)]
pattern = TexasHoldemEnv.__cards2pattern_cards__(h1, keep)[0]
self.assertEqual(pattern[0],"3_1_1")
def test_cards1(self):
"""
"""
handcards1 = [
roomai.common.PokerCard(0, 0),
roomai.common.PokerCard(0, 1)
]
handcards2 = [
roomai.common.PokerCard(3, 1),
roomai.common.PokerCard(3, 2)
]
keepcards = [
roomai.common.PokerCard(0, 2),
roomai.common.PokerCard(0, 3),
roomai.common.PokerCard(2, 0),
roomai.common.PokerCard(2, 1),
roomai.common.PokerCard(3, 3)
]
pattern = TexasHoldemEnv.cards2pattern(handcards2, keepcards)
self.assertEqual(pattern[0], '3_2')
self.assertEqual(pattern[1], False)
self.assertEqual(pattern[2], True)
self.assertEqual(pattern[3], False)
self.assertEqual(pattern[4][0], 3)
self.assertEqual(pattern[4][1], 2)
self.assertEqual(pattern[6][0].point_rank, 3)
self.assertEqual(pattern[6][1].point_rank, 3)
self.assertEqual(pattern[6][2].point_rank, 3)
self.assertEqual(pattern[6][3].point_rank, 2)
self.assertEqual(pattern[6][4].point_rank, 2)
self.assertEqual(pattern[6][0].suit_rank, 1)
self.assertEqual(pattern[6][1].suit_rank, 2)
self.assertEqual(pattern[6][2].suit_rank, 3)
self.assertEqual(pattern[6][3].suit_rank, 0)
self.assertEqual(pattern[6][4].suit_rank, 1)
