def test_ranges(self):
# preflop
t, p, gui_signals, h, logger = init_table(
'tests/screenshots/709250829_PreFlop_0.png')
d = reverse_init(t, h, p, logger)
d.preflop_override(t, logger, h, p)
self.assertEqual('3R1', t.preflop_sheet_name)
preflop_state = CurrentHandPreflopState()
bot_preflop_decision = 'Call'
d = MagicMock()
preflop_state.update_values(t, bot_preflop_decision, h, d)
# river round 2
t, p, gui_signals, h, logger = init_table(
'tests/screenshots/709250829_Flop_0.png')
for abs_pos in range(5):
if t.other_players[abs_pos]['status'] == 1:
sheet_name = preflop_state.get_reverse_sheetname(abs_pos, t, h)
ranges = preflop_state.get_rangecards_from_sheetname(
abs_pos, sheet_name, t, h, p)
self.assertEqual('1', sheet_name)
self.assertEqual(38, len(ranges))
def test_ranges_call_column(self):
# preflop
t, p, gui_signals, h, logger = init_table('tests/screenshots/KQ2.png')
reverse_init(t,h,p,logger)
preflop_state = CurrentHandPreflopState()
bot_preflop_decision = 'Bet'
d = MagicMock()
preflop_state.update_values(t, bot_preflop_decision, h, d)
t, p, gui_signals, h, logger = init_table('tests/screenshots/KQ.png')
for abs_pos in range(5):
if t.other_players[abs_pos]['status'] == 1:
sheet_name = preflop_state.get_reverse_sheetname(abs_pos, t, h)
ranges = preflop_state.get_rangecards_from_sheetname(abs_pos, sheet_name, t, h, p)
self.assertEqual('42R5', sheet_name)
self.assertEqual("Call", preflop_state.range_column_name)
def test_incorrect_second_round(self):
# preflop
t, p, gui_signals, h, logger = init_table('tests/screenshots/3R12.png')
reverse_init(t,h,p,logger)
preflop_state = CurrentHandPreflopState()
bot_preflop_decision = 'Bet'
d = MagicMock()
d = MagicMock()
preflop_state.update_values(t, bot_preflop_decision, h, d)
t, p, gui_signals, h, logger = init_table('tests/screenshots/3R1.png')
reverse_sheet_names=[]
for abs_pos in range(5):
if t.other_players[abs_pos]['status'] == 1:
sheet_name = preflop_state.get_reverse_sheetname(abs_pos, t, h)
ranges = preflop_state.get_rangecards_from_sheetname(abs_pos, sheet_name, t, h, p)
reverse_sheet_names.append(sheet_name)
self.assertEqual('3R1', reverse_sheet_names[0])
self.assertEqual('6R1', reverse_sheet_names[1])
self.assertEqual("Call", preflop_state.range_column_name)
def test_ranges_2nd_round(self):
# preflop
t, p, gui_signals, h, logger = init_table('tests/screenshots/709250829_PreFlop_0.png')
d = reverse_init(t, h, p, logger)
self.assertEqual('3R1', t.preflop_sheet_name)
preflop_state = CurrentHandPreflopState()
bot_preflop_decision = 'Bet'
d = MagicMock()
preflop_state.update_values(t, bot_preflop_decision, h, d)
# flop after 2nd round preflop
t, p, gui_signals, h, logger = init_table('tests/screenshots/709250829_Flop_0.png')
for abs_pos in range(5):
if t.other_players[abs_pos]['status'] == 1:
sheet_name = preflop_state.get_reverse_sheetname(abs_pos, t, h)
ranges = preflop_state.get_rangecards_from_sheetname(abs_pos, sheet_name, t, h, p)
self.assertEqual('12R3', sheet_name)