def testRemoveWall(self): maze = Maze(4, 4, True) # remove a wall from first cell maze.removeWallBetween(0, 1) self.assertCountEqual(maze.getConnectionsOfCellAtIndex(0), [1]) # remove another wall from first cell maze.removeWallBetween(0, 4) self.assertCountEqual(maze.getConnectionsOfCellAtIndex(0), [1, 4])
def testAddWallAgain(self): maze = Maze(4, 4, False) # add a wall from first cell maze.addWallBetween(0, 1) self.assertCountEqual(maze.getConnectionsOfCellAtIndex(0), [4]) # add another wall from first cell maze.addWallBetween(0, 4) self.assertCountEqual(maze.getConnectionsOfCellAtIndex(0), [])
def testFullyConnectedInit(self): maze = Maze(4, 4, False) self.assertIsInstance(maze, Maze) self.assertCountEqual(maze.getConnectionsOfCellAtIndex(0), [1, 4]) self.assertCountEqual(maze.getConnectionsOfCellAtIndex(5), [1, 4, 6, 9])