From d1f7e1b13fa6202fb1d3716362eb6d3293ac7a54 Mon Sep 17 00:00:00 2001
From: Ken Kinder <ken+github@kkinder.com>
Date: Sat, 7 Sep 2024 17:14:19 +0100
Subject: [PATCH] Switch test_default_id_generator.py to pytest

re #61
---
 tests/unittests/test_default_id_generator.py | 30 +++++++++-----------
 1 file changed, 14 insertions(+), 16 deletions(-)

diff --git a/tests/unittests/test_default_id_generator.py b/tests/unittests/test_default_id_generator.py
index 1aa241d..7538a41 100644
--- a/tests/unittests/test_default_id_generator.py
+++ b/tests/unittests/test_default_id_generator.py
@@ -1,25 +1,23 @@
-import unittest
 import itertools
+import pytest
 from puepy.application import DefaultIdGenerator
 
 
-class TestDefaultIdGenerator(unittest.TestCase):
-    def setUp(self):
-        self.id_generator = DefaultIdGenerator("test")
+class TestDefaultIdGenerator:
 
-    def test_init(self):
-        self.assertEqual(self.id_generator.prefix, "test")
-        self.assertIsInstance(self.id_generator.counter, itertools.count)
+    @pytest.fixture
+    def id_generator(self):
+        return DefaultIdGenerator("test")
+
+    def test_init(self, id_generator):
+        assert id_generator.prefix == "test"
+        assert isinstance(id_generator.counter, itertools.count)
 
     def test_int_to_base36(self):
         num = 10
-        self.assertEqual(DefaultIdGenerator._int_to_base36(num), "a")
-
-    def test_call(self):
-        ids = [self.id_generator.get_id_for_element(None) for _ in range(100)]
-        self.assertEqual(ids[0], "test0")
-        self.assertEqual(ids[99], "test2r")
-
+        assert DefaultIdGenerator._int_to_base36(num) == "a"
 
-if __name__ == "__main__":
-    unittest.main()
+    def test_call(self, id_generator):
+        ids = [id_generator.get_id_for_element(None) for _ in range(100)]
+        assert ids[0] == "test0"
+        assert ids[99] == "test2r"