fix the cell pydantic cause the program was too long

src/amaz_lib/Cell.py

@@ -1,8 +1,10 @@
-from pydantic import BaseModel, Field
+from dataclasses import dataclass
 
 
-class Cell(BaseModel):
+@dataclass
-    value: int = Field(ge=0, le=15)
+class Cell:
+    def __init__(self, value: int) -> None:
+        self.value = value
 
     def __str__(self) -> str:
         return hex(self.value).removeprefix("0x").upper()

tests/test_MazeGenerator.py

@@ -1,19 +1,14 @@
 import numpy
 from amaz_lib.MazeGenerator import DepthFirstSearch
-from amaz_lib.MazeGenerator import Kruskal
 
 
 class TestMazeGenerator:
 
-    def test_kruskal_output_shape() -> None:
+    def test_generator(self) -> None:
-        generator = Kruskal().generator(10, 10)
+        w_h = (300, 300)
         maze = numpy.array([])
+        generator = DepthFirstSearch().generator(*w_h)
         for output in generator:
             maze = output
 
-        assert maze.shape == (10, 10)
+        assert maze.shape == w_h
-
-    def test_generator(self) -> None:
-        maze = numpy.array([])
-        maze = DepthFirstSearch.generator(10, 10)
-        assert maze.shape == (10, 10)