Wall.py

import numpy as np


class Wall:

    def __init__(self, initial_state):
        self.state = initial_state
        self.goal = initial_state[1, 1]

    def peek_left_col(self, col):
        return 9 - (np.count_nonzero(self.state[:, 1] == self.goal)
                    + np.count_nonzero(self.state[:, 2] == self.goal)
                    + np.count_nonzero(col == self.goal))

    def peek_right_col(self, col):
        return 9 - (np.count_nonzero(self.state[:, 0] == self.goal)
                    + np.count_nonzero(self.state[:, 1] == self.goal)
                    + np.count_nonzero(col == self.goal))

    def peek_upper_row(self, row):
        return 9 - (np.count_nonzero(self.state[1, :] == self.goal)
                    + np.count_nonzero(self.state[2, :] == self.goal)
                    + np.count_nonzero(row == self.goal))

    def peek_lower_row(self, row):
        return 9 - (np.count_nonzero(self.state[0, :] == self.goal)
                    + np.count_nonzero(self.state[1, :] == self.goal)
                    + np.count_nonzero(row == self.goal))

    def swap_upper_row(self, row):
        previous = self.get_upper_row()
        self.state[0, :] = row
        return previous

    def swap_lower_row(self, row):
        previous = self.get_lower_row()
        self.state[2, :] = row
        return previous

    def swap_left_column(self, column):
        previous = self.get_left_column()
        self.state[:, 0] = column
        return previous

    def swap_right_column(self, column):
        previous = self.get_right_column()
        self.state[:, 2] = column
        return previous

    def get_upper_row(self):
        return self.state[0, :].copy()

    def get_lower_row(self):
        return self.state[2, :].copy()

    def get_left_column(self):
        return self.state[:, 0].copy()

    def get_right_column(self):
        return self.state[:, 2].copy()

    def missing_blocks(self):
        return 9 - np.count_nonzero(self.state == self.goal)