use decorator to parametrize
This commit is contained in:
parent
5ec2909b26
commit
c11156fb80
1 changed files with 5 additions and 9 deletions
|
@ -1,15 +1,11 @@
|
|||
from numpy.testing import assert_allclose
|
||||
|
||||
import pytest
|
||||
from logistic import f
|
||||
|
||||
|
||||
def test_f_corner_cases():
|
||||
@pytest.mark.parametrize('x,r,expected', [(0, 1.1, 0),(1, 3.7, 0)])
|
||||
def test_f_corner_cases(x, r, expected):
|
||||
# Test cases are (x, r, expected)
|
||||
cases = [
|
||||
(0, 1.1, 0),
|
||||
(1, 3.7, 0),
|
||||
]
|
||||
for x, r, expected in cases:
|
||||
result = f(x, r)
|
||||
assert_allclose(result, expected)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue