Skip to content

Commit 60ad958

Browse files
committed
Added type hints for tests
1 parent 20c3200 commit 60ad958

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

maths/polynomials/legendre.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,29 +23,29 @@ def legendre(n: int) -> [float]:
2323
return legendre_polynomial.deriv(n).coef.tolist()
2424

2525

26-
def test_legendre_0():
26+
def test_legendre_0() -> None:
2727
"""Test the 0th Legendre polynomial."""
2828
assert legendre(0) == [1.0], "The 0th Legendre polynomial should be [1.0]"
2929

3030

31-
def test_legendre_1():
31+
def test_legendre_1() -> None:
3232
"""Test the 1st Legendre polynomial."""
3333
assert legendre(1) == [0.0, 1.0], "The 1st Legendre polynomial should be [0.0, 1.0]"
3434

3535

36-
def test_legendre_2():
36+
def test_legendre_2() -> None:
3737
"""Test the 2nd Legendre polynomial."""
3838
assert legendre(2) == [-0.5, 0.0, 1.5]
3939
"The 2nd Legendre polynomial should be [-0.5, 0.0, 1.5]"
4040

4141

42-
def test_legendre_3():
42+
def test_legendre_3() -> None:
4343
"""Test the 3rd Legendre polynomial."""
4444
assert legendre(3) == [0.0, -1.5, 0.0, 2.5]
4545
"The 3rd Legendre polynomial should be [0.0, -1.5, 0.0, 2.5]"
4646

4747

48-
def test_legendre_4():
48+
def test_legendre_4() -> None:
4949
"""Test the 4th Legendre polynomial."""
5050
assert legendre(4) == pytest.approx([0.375, 0.0, -3.75, 0.0, 4.375])
5151
"The 4th Legendre polynomial should be [0.375, 0.0, -3.75, 0.0, 4.375]"

0 commit comments

Comments
 (0)