def _test_from_files(self, in_file_path, out_file_path): with open(in_file_path) as f: b_in = Board.from_file(f) with open(out_file_path) as f: b_out = Board.from_file(f) solve_board(b_in) self.assertEquals(b_in, b_out, "boards don't match (in=%s, out=%s)" % (in_file_path, out_file_path))
def main(): if not 2 <= len(sys.argv) <= 3: usage() sys.exit(-1) with open(sys.argv[1]) as f: b_in = Board.from_file(f) if len(sys.argv) == 3: with open(sys.argv[2]) as f: b_exp = Board.from_file(f) else: b_exp = None print '\nInput:\n\n', b_in, '\n' solve_board(b_in) print '\nSolution:\n\n', b_in, '\n' if b_exp: assert b_in == b_exp, 'board does not match expected result'