starting mypy with maze

This commit is contained in:
maoake
2026-03-31 22:31:48 +02:00
parent b1eda06fa5
commit 40e25757c7
+8 -5
View File
@@ -1,16 +1,16 @@
from dataclasses import dataclass from dataclasses import dataclass
from numpy.typing import NDArray
import numpy from typing import Optional, Any
@dataclass @dataclass
class Maze: class Maze:
maze: numpy.ndarray maze: Optional[NDArray[Any]] = None
def get_maze(self) -> numpy.ndarray | None: def get_maze(self) -> Optional[NDArray[Any]]:
return self.maze return self.maze
def set_maze(self, new_maze: numpy.ndarray) -> None: def set_maze(self, new_maze: NDArray[Any]) -> None:
self.maze = new_maze self.maze = new_maze
def __str__(self) -> str: def __str__(self) -> str:
@@ -24,6 +24,9 @@ class Maze:
return res return res
def ascii_print(self) -> None: def ascii_print(self) -> None:
if self.maze is None:
print("None")
return
for cell in self.maze[0]: for cell in self.maze[0]:
print("_", end="") print("_", end="")
if cell.get_north(): if cell.get_north():