def test_no_bets(self): logic = Game_Helpers() history = np.zeros(1300) history = history.reshape(25, 13, 4) minraise_arr, call_size_arr, check = logic.get_current_raise( history, 'FLOP') minraise_expected = np.zeros(50, dtype=int) minraise_expected[1:] = 1 call_expected = np.zeros(50, dtype=int) check_expected = True self.assertEqual(minraise_expected.tolist(), minraise_arr.tolist()) self.assertEqual(call_expected.tolist(), call_size_arr.tolist()) self.assertEqual(check_expected, check)
def test_blinds_and_underraise(self): logic = Game_Helpers() history = np.zeros(1300) history = history.reshape(25, 13, 4) history[0, 0, 0:1] = 1 history[1, 0, 0:2] = 1 history[2, 0, 0:3] = 1 minraise_arr, call_size_arr, check = logic.get_current_raise( history, 'FLOP') minraise_expected = np.zeros(50, dtype=int) minraise_expected[3:] = 1 call_expected = np.zeros(50, dtype=int) call_expected[2] = 1 check_expected = False self.assertEqual(call_expected.tolist(), call_size_arr.tolist()) self.assertEqual(minraise_expected.tolist(), minraise_arr.tolist()) self.assertEqual(check_expected, check)
def test_check_preflop(self): # 1 SB -> 2SB -> 50 SB logic = Game_Helpers() history = np.zeros(1300) history = history.reshape(25, 13, 4) history[0, 0, 0:1] = 1 history[1, 0, 0:2] = 1 history[2, 0, 0:2] = 1 minraise_arr, call_size_arr, check = logic.get_current_raise( history, 'PREFLOP') minraise_expected = np.zeros(50, dtype=int) minraise_expected[3:] = 1 call_expected = np.zeros(50, dtype=int) check_expected = True self.assertEqual(minraise_expected.tolist(), minraise_arr.tolist()) self.assertEqual(call_expected.tolist(), call_size_arr.tolist()) self.assertEqual(check_expected, check)
def test_allin_50sbs_first_action(self): # 1 SB -> 2SB -> 50 SB logic = Game_Helpers() history = np.zeros(1300) history = history.reshape(25, 13, 4) history[0, 0, 0:1] = 1 history[1, 0, 0:2] = 1 history[2, 0:12, 0:4] = 1 history[2, 12, 0:2] = 1 minraise_arr, call_size_arr, check = logic.get_current_raise( history, 'FLOP') minraise_expected = np.zeros(50, dtype=int) call_expected = np.zeros(50, dtype=int) call_expected[49] = 1 check_expected = False self.assertEqual(minraise_expected.tolist(), minraise_arr.tolist()) self.assertEqual(call_expected.tolist(), call_size_arr.tolist()) self.assertEqual(check_expected, check)
def test_normal_raise(self): #1 SB -> 2SB -> 6SB -> minraise = 10 SB logic = Game_Helpers() history = np.zeros(1300) history = history.reshape(25, 13, 4) history[0, 0, 0:1] = 1 history[1, 0, 0:2] = 1 history[2, 0, 0:4] = 1 history[2, 1, 0:2] = 1 minraise_arr, call_size_arr, check = logic.get_current_raise( history, 'FLOP') minraise_expected = np.zeros(50, dtype=int) minraise_expected[9:] = 1 call_expected = np.zeros(50, dtype=int) call_expected[5] = 1 check_expected = False self.assertEqual(minraise_expected.tolist(), minraise_arr.tolist()) self.assertEqual(call_expected.tolist(), call_size_arr.tolist()) self.assertEqual(check_expected, check)
def test_full_allin(self): logic = Game_Helpers() history = np.zeros(1300) history = history.reshape(25, 13, 4) #Small Blind history[0, 0, 0:1] = 1 #Big Blind history[1, 0, 0:2] = 1 history[2, 0:12, :] = 1 history[2, 12, 0:2] = 1 minraise_arr, call_size_arr, check = logic.get_current_raise( history, 'FLOP') minraise_expected = np.zeros(50, dtype=int) call_expected = np.zeros(50, dtype=int) call_expected[49] = 1 check_expected = False self.assertEqual(minraise_expected.tolist(), minraise_arr.tolist()) self.assertEqual(call_expected.tolist(), call_size_arr.tolist()) self.assertEqual(check_expected, check)