def test_h1():
puzzle = Puzzle(2, [1, 2, 3, 4, 0, 6, 7, 5])
assert puzzle.get_heuristic('h1') == 2
puzzle = Puzzle(2, [1, 2, 3, 4, 6, 0, 7, 5])
assert puzzle.get_heuristic('h1') == 3
puzzle = Puzzle(4, [1, 2, 7, 4, 6, 3, 11, 5, 8, 0, 10, 9],
goal_setups=[[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 0]])
assert puzzle.get_heuristic('h1') == 9
def test_h2():
puzzle = Puzzle(2, [1, 2, 3, 4, 5, 6, 0, 7])
assert puzzle.get_heuristic('h2') == 1
# puzzle = Puzzle(2, [1, 2, 3, 4, 6, 0, 7, 5])
# assert puzzle.get_heuristic('h2') == 3
#
# puzzle = Puzzle(4, [1, 2, 7, 4, 6, 3, 11, 5, 8, 0, 10, 9], goal_setups=[[1,2,3,4,5,6,7,8,9,10,11,0]])
# assert puzzle.get_heuristic('h2') == 9
puzzle = Puzzle(3, [5, 0, 8, 4, 2, 1, 7, 3, 6],
goal_setups=[[1, 2, 3, 4, 5, 6, 7, 8, 0]])
assert puzzle.get_heuristic('h2') == 17
from argparse import ArgumentParser
from resource import getrusage, RUSAGE_SELF
from src.puzzle import Puzzle
if __name__ == "__main__":
try:
parser = ArgumentParser()
parser.add_argument("solver", help="algorithm (bfs | dfs)")
parser.add_argument("board", help="board string (0,1,2,3...)")
args = parser.parse_args()
puzzle = Puzzle(tiles=args.board, algorithm=args.solver)
results = puzzle.solve()
with open('output.txt', 'w+') as f:
f.write(f"path_to_goal: {puzzle.actions}\n")
f.write(f"cost_of_path: {puzzle.path_cost}\n")
f.write(f"nodes_expanded: {puzzle.nodes_expanded}\n")
f.write(f"fringe_size: {puzzle.fringe_size}\n")
f.write(f"max_fringe_size: {puzzle.max_fringe_size}\n")
f.write(f"search_depth: {puzzle.search_depth}\n")
f.write(f"max_search_depth: {puzzle.max_search_depth}\n")
f.write(f"running_time: {puzzle.running_time}\n")
f.write(f"max_ram_usage: {getrusage(RUSAGE_SELF)[2]/(1024**2)}\n")
except TypeError as e:
print(e)
exit(1)
except RuntimeError as e:
print(e)
exit(1)
def run():
puzzle = Puzzle()
args = ArgParser().parse()
if args.file and args.size:
puzzle.common_error(
"flag --file and --size must be specify separately")
if args.size and (args.size < 2 or args.size == 0):
puzzle.critical_error("size should be >= 2")
if args.file:
puzzle.read(args.file)
puzzle.validate()
else:
if args.size:
puzzle.size = args.size
puzzle.get_heuristic_type(args.htype)
puzzle.generate(args.file)
puzzle.print()
node = puzzle.solver()
puzzle.count_moves(node, args.path)
print('\nSolution is:')
puzzle.print(node=node)
puzzle.output_required()
def puzzle(dict_puzzle):
screen = Puzzle(c.GRID_WIDTH, c.GRID_HEIGHT, 100)
screen.squares = dict_puzzle
print(dict_puzzle)
screen.blank_square = (MAX_X, MAX_Y)
return screen
def test_3x3_7():
puzzle = Puzzle()
htype = "m"
file = f"{testpath}test_3x3_7.txt"
puzzle.read(file)
puzzle.validate()
puzzle.get_heuristic_type(htype)
puzzle.generate(file)
node = puzzle.solver()
assert node.grid == node.solved_puzzle
def test_error_empty():
puzzle = Puzzle()
file = f"{testpath}test_error_empty.txt"
with pytest.raises(SystemExit):
puzzle.read(file)
puzzle.validate()
def test_move_down():
puzzle = Puzzle(2, [1, 2, 3, 0, 4, 6, 7, 5])
assert puzzle.move_down()[0].current_setup == [1, 2, 3, 5, 4, 6, 7, 0]
puzzle = Puzzle(2, [1, 0, 3, 4, 6, 2, 7, 5])
assert puzzle.move_down()[0].current_setup == [1, 2, 3, 4, 6, 0, 7, 5]
puzzle = Puzzle(3, [1, 2, 0, 4, 6, 3, 7, 5, 8])
assert puzzle.move_down()[0].current_setup == [1, 2, 3, 4, 6, 0, 7, 5, 8]
puzzle = Puzzle(3, [1, 2, 3, 4, 0, 8, 7, 6, 5])
assert puzzle.move_down()[0].current_setup == [1, 2, 3, 4, 6, 8, 7, 0, 5]
return
def test_possible_moves_2x4():
setup_1 = [0, 2, 3, 1, 4, 6, 7, 5]
setup_2 = [1, 2, 3, 0, 4, 6, 7, 5]
setup_3 = [1, 2, 3, 4, 0, 6, 7, 5]
setup_4 = [1, 2, 3, 4, 5, 6, 7, 0]
setup_5 = [1, 2, 3, 4, 5, 0, 7, 6]
puzzle_1 = Puzzle(2, setup_1)
possible_moves_p1 = [
'move_wrapping', 'move_diagonal_short', 'move_diagonal_long',
'move_right', 'move_down'
]
assert puzzle_1.possible_moves() == possible_moves_p1
puzzle_2 = Puzzle(2, setup_2)
possible_moves_p2 = [
'move_wrapping', 'move_diagonal_short', 'move_diagonal_long',
'move_left', 'move_down'
]
assert puzzle_2.possible_moves() == possible_moves_p2
puzzle_3 = Puzzle(2, setup_3)
possible_moves_p3 = [
'move_wrapping', 'move_diagonal_short', 'move_diagonal_long',
'move_right', 'move_up'
]
assert puzzle_3.possible_moves() == possible_moves_p3
puzzle_4 = Puzzle(2, setup_4)
possible_moves_p4 = [
'move_wrapping', 'move_diagonal_short', 'move_diagonal_long',
'move_left', 'move_up'
]
assert puzzle_4.possible_moves() == possible_moves_p4
puzzle_5 = Puzzle(2, setup_5)
possible_moves_p5 = ['move_right', 'move_left', 'move_up']
assert puzzle_5.possible_moves() == possible_moves_p5
return
def test_possible_moves_other_sizes():
setup_6 = [1, 2, 3, 4, 5, 6, 7, 0, 9, 10, 11, 8]
setup_7 = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 0]
puzzle_6 = Puzzle(2, setup_6)
possible_moves_p6 = ['move_right', 'move_left', 'move_up']
assert puzzle_6.possible_moves() == possible_moves_p6
puzzle_7 = Puzzle(3, setup_6)
possible_moves_p7 = ['move_left', 'move_down', 'move_up']
assert puzzle_7.possible_moves() == possible_moves_p7
puzzle_8 = Puzzle(4, setup_6)
possible_moves_p8 = ['move_right', 'move_left', 'move_down', 'move_up']
assert puzzle_8.possible_moves() == possible_moves_p8
puzzle_9 = Puzzle(4, setup_7)
possible_moves_p9 = [
'move_wrapping', 'move_diagonal_short', 'move_diagonal_long',
'move_left', 'move_up'
]
assert puzzle_9.possible_moves() == possible_moves_p9
return
def test_move_diagonal_long():
puzzle = Puzzle(2, [1, 2, 3, 4, 6, 5, 7, 0])
assert puzzle.move_diagonal_long()[0].current_setup == [
0, 2, 3, 4, 6, 5, 7, 1
]
assert puzzle.move_diagonal_long()[0].rows == 2
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(2, [1, 2, 3, 4, 0, 5, 7, 6])
assert puzzle.move_diagonal_long()[0].current_setup == [
1, 2, 3, 0, 4, 5, 7, 6
]
assert puzzle.move_diagonal_long()[0].rows == 2
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(2, [1, 2, 3, 0, 6, 5, 7, 4])
assert puzzle.move_diagonal_long()[0].current_setup == [
1, 2, 3, 6, 0, 5, 7, 4
]
assert puzzle.move_diagonal_long()[0].rows == 2
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(2, [0, 2, 3, 1, 6, 5, 7, 4])
assert puzzle.move_diagonal_long()[0].current_setup == [
4, 2, 3, 1, 6, 5, 7, 0
]
assert puzzle.move_diagonal_long()[0].rows == 2
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(3, [1, 2, 0, 4, 6, 3, 7, 5, 8])
assert puzzle.move_diagonal_long()[0].current_setup == [
1, 2, 7, 4, 6, 3, 0, 5, 8
]
assert puzzle.move_diagonal_long()[0].rows == 3
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(3, [1, 2, 7, 4, 6, 3, 11, 5, 8, 9, 10, 0])
assert puzzle.move_diagonal_long()[0].current_setup == [
0, 2, 7, 4, 6, 3, 11, 5, 8, 9, 10, 1
]
assert puzzle.move_diagonal_long()[0].rows == 3
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(4, [1, 2, 7, 4, 6, 3, 11, 5, 8, 9, 10, 0])
assert puzzle.move_diagonal_long()[0].current_setup == [
0, 2, 7, 4, 6, 3, 11, 5, 8, 9, 10, 1
]
assert puzzle.move_diagonal_long()[0].rows == 4
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(4, [1, 2, 0, 4, 6, 3, 11, 5, 7, 8, 10, 9])
assert puzzle.move_diagonal_long()[0].current_setup == [
1, 2, 8, 4, 6, 3, 11, 5, 7, 0, 10, 9
]
assert puzzle.move_diagonal_long()[0].rows == 4
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
puzzle = Puzzle(4, [1, 2, 7, 4, 6, 3, 11, 5, 8, 0, 10, 9])
assert puzzle.move_diagonal_long()[0].current_setup == [
1, 2, 0, 4, 6, 3, 11, 5, 8, 7, 10, 9
]
assert puzzle.move_diagonal_long()[0].rows == 4
assert puzzle.move_diagonal_long()[1] == 'move_diagonal_long'
assert puzzle.move_diagonal_long()[2] == 3
return
def test_move_diagonal_short():
puzzle = Puzzle(2, [1, 2, 3, 4, 6, 5, 7, 0])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 0, 4, 6, 5, 7, 3
]
assert puzzle.move_diagonal_short()[0].rows == 2
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(2, [1, 2, 3, 4, 0, 5, 7, 6])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 0, 3, 4, 2, 5, 7, 6
]
assert puzzle.move_diagonal_short()[0].rows == 2
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(2, [1, 2, 3, 0, 6, 5, 7, 4])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 3, 7, 6, 5, 0, 4
]
assert puzzle.move_diagonal_short()[0].rows == 2
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(2, [0, 2, 3, 1, 6, 5, 7, 4])
assert puzzle.move_diagonal_short()[0].current_setup == [
5, 2, 3, 1, 6, 0, 7, 4
]
assert puzzle.move_diagonal_short()[0].rows == 2
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(3, [1, 2, 0, 4, 6, 3, 7, 5, 8])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 6, 4, 0, 3, 7, 5, 8
]
assert puzzle.move_diagonal_short()[0].rows == 3
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(3, [1, 2, 7, 4, 6, 3, 11, 5, 8, 9, 10, 0])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 7, 4, 6, 3, 0, 5, 8, 9, 10, 11
]
assert puzzle.move_diagonal_short()[0].rows == 3
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(4, [1, 2, 7, 4, 6, 3, 11, 5, 8, 9, 10, 0])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 7, 4, 6, 3, 11, 0, 8, 9, 10, 5
]
assert puzzle.move_diagonal_short()[0].rows == 4
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(4, [1, 2, 7, 4, 6, 3, 11, 5, 8, 0, 10, 9])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 7, 4, 6, 3, 11, 0, 8, 5, 10, 9
]
assert puzzle.move_diagonal_short()[0].rows == 4
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
puzzle = Puzzle(4, [1, 2, 0, 4, 6, 3, 11, 5, 8, 7, 10, 9])
assert puzzle.move_diagonal_short()[0].current_setup == [
1, 2, 6, 4, 0, 3, 11, 5, 8, 7, 10, 9
]
assert puzzle.move_diagonal_short()[0].rows == 4
assert puzzle.move_diagonal_short()[1] == 'move_diagonal_short'
assert puzzle.move_diagonal_short()[2] == 3
return
def test_move_wrapping():
puzzle = Puzzle(2, [1, 2, 3, 4, 6, 5, 7, 0])
assert puzzle.move_wrapping()[0].current_setup == [1, 2, 3, 4, 0, 5, 7, 6]
assert puzzle.move_wrapping()[0].rows == 2
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
puzzle = Puzzle(2, [1, 2, 3, 4, 0, 5, 7, 6])
assert puzzle.move_wrapping()[0].current_setup == [1, 2, 3, 4, 6, 5, 7, 0]
assert puzzle.move_wrapping()[0].rows == 2
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
puzzle = Puzzle(2, [1, 2, 3, 0, 6, 5, 7, 4])
assert puzzle.move_wrapping()[0].current_setup == [0, 2, 3, 1, 6, 5, 7, 4]
assert puzzle.move_wrapping()[0].rows == 2
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
puzzle = Puzzle(2, [0, 2, 3, 1, 6, 5, 7, 4])
assert puzzle.move_wrapping()[0].current_setup == [1, 2, 3, 0, 6, 5, 7, 4]
assert puzzle.move_wrapping()[0].rows == 2
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
puzzle = Puzzle(3, [1, 2, 0, 4, 6, 3, 7, 5, 8])
assert puzzle.move_wrapping()[0].current_setup == [
0, 2, 1, 4, 6, 3, 7, 5, 8
]
assert puzzle.move_wrapping()[0].rows == 3
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
puzzle = Puzzle(3, [1, 2, 7, 4, 6, 3, 0, 5, 8])
assert puzzle.move_wrapping()[0].current_setup == [
1, 2, 7, 4, 6, 3, 8, 5, 0
]
assert puzzle.move_wrapping()[0].rows == 3
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
puzzle = Puzzle(3, [1, 2, 3, 4, 8, 5, 7, 6, 0])
assert puzzle.move_wrapping()[0].current_setup == [
1, 2, 3, 4, 8, 5, 0, 6, 7
]
assert puzzle.move_wrapping()[0].rows == 3
assert puzzle.move_wrapping()[1] == 'move_wrapping'
assert puzzle.move_wrapping()[2] == 2
return
def test_move_left():
puzzle = Puzzle(2, [1, 2, 3, 4, 6, 0, 7, 5])
assert puzzle.move_left()[0].current_setup == [1, 2, 3, 4, 0, 6, 7, 5]
assert puzzle.move_left()[0].rows == 2
assert puzzle.move_left()[1] == 'move_left'
assert puzzle.move_left()[2] == 1
puzzle = Puzzle(2, [1, 3, 0, 4, 6, 2, 7, 5])
assert puzzle.move_left()[0].current_setup == [1, 0, 3, 4, 6, 2, 7, 5]
assert puzzle.move_left()[0].rows == 2
assert puzzle.move_left()[1] == 'move_left'
assert puzzle.move_left()[2] == 1
puzzle = Puzzle(2, [1, 0, 3, 4, 6, 2, 7, 5])
assert puzzle.move_left()[0].current_setup == [0, 1, 3, 4, 6, 2, 7, 5]
assert puzzle.move_left()[0].rows == 2
assert puzzle.move_left()[1] == 'move_left'
assert puzzle.move_left()[2] == 1
puzzle = Puzzle(2, [5, 1, 3, 4, 6, 2, 7, 0])
assert puzzle.move_left()[0].current_setup == [5, 1, 3, 4, 6, 2, 0, 7]
assert puzzle.move_left()[0].rows == 2
assert puzzle.move_left()[1] == 'move_left'
assert puzzle.move_left()[2] == 1
puzzle = Puzzle(3, [1, 2, 0, 4, 6, 3, 7, 5, 8])
assert puzzle.move_left()[0].current_setup == [1, 0, 2, 4, 6, 3, 7, 5, 8]
assert puzzle.move_left()[0].rows == 3
assert puzzle.move_left()[1] == 'move_left'
assert puzzle.move_left()[2] == 1
puzzle = Puzzle(3, [1, 2, 3, 4, 8, 0, 7, 6, 5])
assert puzzle.move_left()[0].current_setup == [1, 2, 3, 4, 0, 8, 7, 6, 5]
assert puzzle.move_left()[0].rows == 3
assert puzzle.move_left()[1] == 'move_left'
assert puzzle.move_left()[2] == 1
return