From bf27ee6bb939d91ad25901d304088a50f2ca7159 Mon Sep 17 00:00:00 2001
From: Dan Anglin <daangling@gmail.com>
Date: Sun, 18 Feb 2024 09:10:34 +0000
Subject: [PATCH] tests: add fixes for tests

---
 maze.py  | 5 ++++-
 tests.py | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/maze.py b/maze.py
index 981b895..187b10d 100644
--- a/maze.py
+++ b/maze.py
@@ -121,7 +121,10 @@ class Maze:
         else:
             self._graphics.clear_all()
             self._reset_cell_grid()
-        self._draw_cell_grid()
+
+        if self._graphics:
+            self._draw_cell_grid()
+
         self._open_entrance_and_exit()
         self._break_walls_r(MazePosition(
             i=0,
diff --git a/tests.py b/tests.py
index b3a1a3c..b6d59b8 100644
--- a/tests.py
+++ b/tests.py
@@ -39,6 +39,7 @@ class Tests(unittest.TestCase):
                 None,
                 None,
             )
+            m.generate()
             self.assertEqual(
                 len(m._cell_grid),
                 case["height"],