Skip to content

Commit 9696330

Browse files
authored
Update test_factorial.py
1 parent 137dda1 commit 9696330

File tree

1 file changed

+10
-13
lines changed

1 file changed

+10
-13
lines changed

recursion/tests/test_factorial.py

Lines changed: 10 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,15 @@
1-
import unittest
2-
from recursion.factorial import factorial
1+
import pytest
32

3+
from recursion.factorial import factorial
44

5-
class TestFactorial(unittest.TestCase):
6-
def test_factorial_valid_inputs(self):
7-
self.assertEqual(factorial(0), 1)
8-
self.assertEqual(factorial(1), 1)
9-
self.assertEqual(factorial(5), 120)
10-
self.assertEqual(factorial(10), 3628800)
115

12-
def test_factorial_invalid_input(self):
13-
with self.assertRaises(ValueError):
14-
factorial(-1)
6+
def test_factorial_valid_inputs() -> None:
7+
assert factorial(0) == 1
8+
assert factorial(1) == 1
9+
assert factorial(5) == 120
10+
assert factorial(10) == 3628800
1511

1612

17-
if __name__ == "__main__":
18-
unittest.main()
13+
def test_factorial_invalid_input() -> None:
14+
with pytest.raises(ValueError):
15+
factorial(-1)

0 commit comments

Comments
 (0)