uvx --python=3.12 unittest2pytest --write test_factorial.py

Author: cclauss

maths/test_factorial.py

@@ -1,23 +1,37 @@
-import unittest
-from factorial import factorial
+# /// script
+# requires-python = ">=3.13"
+# dependencies = [
+#     "pytest",
+# ]
+# ///
 
+import pytest
 
-class TestFactorial(unittest.TestCase):
-    def test_zero(self):
-        self.assertEqual(factorial(0), 1)
+from maths.factorial import factorial, factorial_recursive
 
-    def test_positive_integers(self):
-        self.assertEqual(factorial(1), 1)
-        self.assertEqual(factorial(5), 120)
-        self.assertEqual(factorial(7), 5040)
 
-    def test_large_number(self):
-        self.assertEqual(factorial(10), 3628800)
+@pytest.mark.parametrize("function", [factorial, factorial_recursive])
+def test_zero(function):
+    assert function(0) == 1
 
-    def test_negative_number(self):
-        with self.assertRaises(ValueError):
-            factorial(-3)
+
+@pytest.mark.parametrize("function", [factorial, factorial_recursive])
+def test_positive_integers(function):
+    assert function(1) == 1
+    assert function(5) == 120
+    assert function(7) == 5040
+
+
+@pytest.mark.parametrize("function", [factorial, factorial_recursive])
+def test_large_number(function):
+    assert function(10) == 3628800
+
+
+@pytest.mark.parametrize("function", [factorial, factorial_recursive])
+def test_negative_number(function):
+    with pytest.raises(ValueError):
+        function(-3)
 
 
 if __name__ == "__main__":
-    unittest.main()
+    pytest.main(["-v", __file__])