Solution: test more exceptions
def fib(n):
if n < 1 or int(n) != n:
raise ValueError(f'Invalid parameter {n}')
a, b = 1, 1
for _ in range(1, n):
a, b = b, a+b
return a
import pytest
from fibonacci import fib
def test_fib():
assert fib(10) == 55
def test_fib_negative():
with pytest.raises(Exception) as err:
fib(-1)
assert err.type == ValueError
assert str(err.value) == 'Invalid parameter -1'
def test_fib_negative_again():
with pytest.raises(ValueError) as err:
fib(-1)
assert str(err.value) == 'Invalid parameter -1'
def test_fib_negative_again():
with pytest.raises(ValueError) as err:
fib(3.5)
assert str(err.value) == 'Invalid parameter 3.5'
Output:
============================= test session starts ==============================
platform linux -- Python 3.8.6, pytest-6.1.2, py-1.9.0, pluggy-0.13.1
rootdir: /home/gabor/work/slides/python/examples/pytest/fib7
plugins: flake8-1.0.6, dash-1.17.0
collected 3 items
test_fibonacci.py ... [100%]
============================== 3 passed in 0.01s ===============================