def test_ocr_valu36(): # ok
log = logging.getLogger(__name__)
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
table_scraper.screenshot = Image.open(
os.path.join(os.environ['test_test'], '5.png'))
table_scraper.crop_from_top_left_corner2(select_table=2)
if is_debug:
# table_scraper.screenshot.save('log/pics/table1.png')
file_name = get_dir(os.path.join('log', 'pics', 'table_label.png'))
save_table_rectangle_cv2(table_scraper.screenshot, table_dict,
file_name)
table_scraper.get_pots()
table_scraper.get_players_funds()
table_scraper.get_player_pots()
log.info(f"{table_scraper.total_pot}")
log.info(f"{table_scraper.player_funds}")
log.info(f"{table_scraper.player_pots}")
pytest.assume(table_scraper.total_pot == 12.0)
pytest.assume(
table_scraper.player_funds == [145.5, 100.0, 20.0, 18.0, 187.5, 94.5])
pytest.assume(
table_scraper.player_pots == [-1.0, -1.0, -1.0, -1.0, -1.0, -1.0])
def test_ocr_valu22(): # ok
# init_logger(screenlevel=logging.DEBUG, filename='test-ocr', logdir='log')
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
table_scraper.screenshot = Image.open(
os.path.join(os.environ['test_test'], '2.png'))
table_scraper.crop_from_top_left_corner2(select_table=3)
if is_debug:
# table_scraper.screenshot.save('log/pics/table1.png')
file_name = get_dir(os.path.join('log', 'pics', 'table_label.png'))
save_table_rectangle_cv2(table_scraper.screenshot, table_dict,
file_name)
table_scraper.get_pots()
table_scraper.get_players_funds()
table_scraper.get_player_pots()
log.info(f"{table_scraper.total_pot}")
log.info(f"{table_scraper.player_funds}")
log.info(f"{table_scraper.player_pots}")
pytest.assume(table_scraper.total_pot == 12.5)
pytest.assume(
table_scraper.player_funds == [154.0, 104.5, 90.0, 95.0, 20.0, 190.0])
pytest.assume(table_scraper.player_pots == [
2.0, 0.5, 10.0, -1.0, -1.0, -1.0
]) # [2.0, 0.5, 10.0, -1.0, 4.0, -1.0]
def test_ocr_valu10(): # ok
log = logging.getLogger(__name__)
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
table_scraper.screenshot = Image.open(
os.path.join(os.environ['test_src'], '1.png'))
table_scraper.crop_from_top_left_corner2(select_table=3)
if is_debug:
# table_scraper.screenshot.save('log/pics/table1.png')
file_name = get_dir(os.path.join('log', 'pics', 'table_label.png'))
save_table_rectangle_cv2(table_scraper.screenshot, table_dict,
file_name)
table_scraper.get_pots()
table_scraper.get_players_funds()
table_scraper.get_player_pots()
log.info(f"{table_scraper.total_pot}")
log.info(f"{table_scraper.player_funds}")
log.info(f"{table_scraper.player_pots}")
pytest.assume(table_scraper.total_pot == 5.0) # 73.0
pytest.assume(table_scraper.player_funds == [
25.0, 11.0, 98.0, 188.5, 50.5, 100.0
]) # [29.0, 6.0, 98.0, 108.9, 30.5, 10.0]
pytest.assume(table_scraper.player_pots == [
-1.0, -1.0, 2.0, 2.0, 1.0, -1.0
]) # [-1.0, -1.0, 25.0, -1.0, 1.0, -1.0]
def test_ocr_valu8(): # ok
log = logging.getLogger(__name__)
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
table_scraper.screenshot = Image.open(
os.path.join(os.environ['test_src'], '1.png'))
table_scraper.crop_from_top_left_corner2(select_table=0)
if is_debug:
# table_scraper.screenshot.save('log/pics/table1.png')
file_name = get_dir(os.path.join('log', 'pics', 'table_label.png'))
save_table_rectangle_cv2(table_scraper.screenshot, table_dict,
file_name)
table_scraper.get_pots()
table_scraper.get_players_funds()
table_scraper.get_player_pots()
pytest.assume(table_scraper.total_pot == 12.5)
pytest.assume(
table_scraper.player_funds == [23.0, 100.0, 42.0, 23.0, 97.5,
102.0]) # pass
pytest.assume(
table_scraper.player_pots == [-1.0, -1.0, 6.0, -1.0, -1.0,
-1.0]) # Ghost number
def test_ocr_valu6(): # ok
log = logging.getLogger(__name__)
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
table_scraper.screenshot = Image.open(
os.path.join(os.environ['test_src1'], '9.png'))
table_scraper.crop_from_top_left_corner2(select_table=0)
if is_debug:
# table_scraper.screenshot.save('log/pics/table1.png')
file_name = get_dir(os.path.join('log', 'pics', 'table_label.png'))
save_table_rectangle_cv2(table_scraper.screenshot, table_dict,
file_name)
table_scraper.get_pots()
table_scraper.get_players_funds()
table_scraper.get_player_pots()
log.info(f"{table_scraper.total_pot}")
log.info(f"{table_scraper.player_funds}")
log.info(f"{table_scraper.player_pots}")
# assert table_scraper.total_pot == 4
pytest.assume(table_scraper.total_pot == 4.0) # 0.4 == 4.0 capture err
pytest.assume(
table_scraper.player_funds == [16.5, 23.5, 225.0, 65.5, 101.5, 100.0])
pytest.assume(
table_scraper.player_pots == [-1.0, 0.5, 1.0, 2.5, -1.0, -1.0])
def test_4ocr_valu44(): # ok
log = logging.getLogger(__name__)
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
file_name = get_dir(os.path.join('log', 'pics', 'table_b_16.png'))
table_scraper.screenshot = Image.open(file_name)
# table_scraper.crop_from_top_left_corner2(select_table=0)
if is_debug:
# table_scraper.screenshot.save('log/pics/table1.png')
file_name = get_dir(os.path.join('log', 'pics', 'table_label.png'))
save_table_rectangle_cv2(table_scraper.screenshot, table_dict,
file_name)
table_scraper.get_pots()
table_scraper.get_players_funds()
table_scraper.get_player_pots()
log.info(f"{table_scraper.total_pot}")
log.info(f"{table_scraper.current_round_pot}")
log.info(f"{table_scraper.player_funds}")
log.info(f"{table_scraper.player_pots}")
# pytest.assume(table_scraper.total_pot == 4.0)
# assert table_scraper.total_pot == 4.0
# pytest.assume(table_scraper.player_pots == [-1.0, -1.0, -1.0, 0.5, -1.0, 1.0])
pytest.assume(table_scraper.current_round_pot == 2.0)
def test_get_table_delta1():
p = StrategyHandler()
p.read_strategy()
p.read_strategy()
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
gui_signals = MagicMock()
gui_signals.select_table = 0 # select table
game_logger = GameLogger()
version = 4.21
log_file = get_dir(os.path.join("log", 'delta.txt'))
f = open(log_file, "a")
f.write("New Project\n")
f.close()
count = 0
while True:
t = TableScreenBased(p, table_dict, gui_signals, game_logger, version)
ready = False
while (not ready):
ready = t.take_screenshot(True, p) and \
t.get_top_left_corner(p, gui_signals) and \
t.get_gg_poker()
log.debug(f"Top left corner: {t.tlc}")
log.debug(f"GG Poker : {t.gg_poker}")
x, y = get_table_delta()
update_delta(x, y)
delta_x = 308 + (t.tlc[0] - t.gg_poker[0])
delta_y = 287 + (t.tlc[1] - t.gg_poker[1])
f = open(log_file, "a")
f.write(f"delta_x, delta_y : {delta_x}, {delta_y} ")
f.write("\n")
f.write("---------------------------------------------------")
f.write("\n")
f.close()
log.debug(f"delta_x, delta_y : {delta_x}, {delta_y} ")
file_name = get_dir(os.path.join('log', 'pics', 'table_update_delta.png'))
file_name1 = get_dir(os.path.join('log', 'pics', 'table_default_delta.png'))
file_name2 = get_dir(os.path.join('log', 'pics', 'table_default_origin.png'))
"""
Move table by delta_x = -8 , delta_y = -5 from config.ini
"""
save_table_rectangle_cv2(t.screenshot, table_dict, file_name)
reset_delta()
save_table_rectangle_cv2(t.screenshot, table_dict, file_name1)
t.screenshot.save(file_name2)
count += 1
if count >=1:
reset_delta()
break
return True
def test_get_table_delta_rel2():
p = StrategyHandler()
p.read_strategy()
p.read_strategy()
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
gui_signals = MagicMock()
gui_signals.select_table = 0 # select table
game_logger = GameLogger()
version = 4.21
log_file = get_dir(os.path.join("log", 'delta.txt'))
f = open(log_file, "a")
f.write("New Project\n")
f.close()
count = 0
prefix = 'e'
while True:
t = TableScreenBased(p, table_dict, gui_signals, game_logger, version)
t.screenshot = Image.open(get_dir(os.path.join('log', 'pics', 'table_' + prefix + '0.png')))
t.get_gg_poker_rel()
x = 308 - t.gg_poker_rel[0]
y = 287 - t.gg_poker_rel[1]
update_delta(x, y) # update delta in config.ini
f = open(log_file, "a")
f.write(f"delta_x, delta_y : {x}, {y} ")
f.write("\n")
f.write("---------------------------------------------------")
f.write("\n")
f.close()
log.debug(f"delta_x, delta_y : {x}, {y} ")
file_name = get_dir(os.path.join('log', 'pics', 'table_update_delta.png'))
file_name1 = get_dir(os.path.join('log', 'pics', 'table_default_delta.png'))
"""
Move table by delta_x = -8 , delta_y = -5 from config.ini
"""
save_table_rectangle_cv2(t.screenshot, table_dict, file_name)
# reset delta to default in config.ini
reset_delta()
save_table_rectangle_cv2(t.screenshot, table_dict, file_name1)
count += 1
if count >=1:
break
return True
def test_label_pos0(): # ok
log = logging.getLogger(__name__)
mongo = MongoManager()
table_dict = mongo.get_table('GG_6TABLE')
table_scraper = TableScraper(table_dict)
prefix = 'b_16'
table_scraper.screenshot = Image.open(get_dir(os.path.join('log', 'pics', 'table_' + prefix + '.png')))
for i in [0]:
for j in [0]:
# table_scraper.crop_from_top_left_corner2(select_table=0)
if is_debug:
file_name = get_dir(os.path.join('log', 'pics', 'table_label_pos_' + str(i) + '_' + str(j) + '.png'))
"""
Move table by delta_x = -8 , delta_y = -5 from config.ini
"""
save_table_rectangle_cv2(table_scraper.screenshot, table_dict, file_name)
return True
