def test_se_tardata_sudokus(file_path_base):
try:
image_file_name = file_path_base + '.jpg'
image = load_image(image_file_name)
except:
image_file_name = file_path_base + '.png'
image = load_image(image_file_name)
with open(file_path_base + '.txt', 'rt') as f:
correct_parsing = f.read().strip()
predictions, sudoku, subimage = extract_sudoku(image, classifier(), force=True)
parsed_sudoku = predictions_to_suduko_string(predictions, oneliner=False)
if parsed_sudoku != correct_parsing:
predictions, sudoku, subimage = extract_sudoku(image.rotate(-90), classifier(), force=True)
parsed_sudoku = predictions_to_suduko_string(predictions, oneliner=False)
assert parsed_sudoku == correct_parsing
try:
os.remove(image_file_name)
os.remove(file_path_base + '.txt')
print("Deleted temporary test dir.")
os.removedirs(os.path.dirname(file_path_base))
except:
pass
def test_se_tardata_sudokus(file_path_base):
try:
image_file_name = file_path_base + '.jpg'
image = load_image(image_file_name)
except:
image_file_name = file_path_base + '.png'
image = load_image(image_file_name)
with open(file_path_base + '.txt', 'rt') as f:
correct_parsing = f.read().strip()
predictions, sudoku, subimage = extract_sudoku(image,
classifier(),
force=True)
parsed_sudoku = predictions_to_suduko_string(predictions, oneliner=False)
if parsed_sudoku != correct_parsing:
predictions, sudoku, subimage = extract_sudoku(image.rotate(-90),
classifier(),
force=True)
parsed_sudoku = predictions_to_suduko_string(predictions,
oneliner=False)
assert parsed_sudoku == correct_parsing
try:
os.remove(image_file_name)
os.remove(file_path_base + '.txt')
print("Deleted temporary test dir.")
os.removedirs(os.path.dirname(file_path_base))
except:
pass
def main():
import argparse
parser = argparse.ArgumentParser(description="Running SudokuExtract as a command line tool")
group = parser.add_mutually_exclusive_group()
group.add_argument('-p', '--path', dest='path', type=str, default=None, help="Path to an image to parse.")
group.add_argument('-u', '--url', dest='url', type=str, default=None, help="Url to an image to parse.")
parser.add_argument('--oneliner', action='store_true', help="Print oneliner solution.")
args = parser.parse_args()
if args.path is not None:
image = load_image(args.path)
else:
image = download_image(args.url)
preds, images, subimage = extract_sudoku(image)
sudoku_string = predictions_to_suduko_string(preds, args.oneliner)
print(sudoku_string)
