Minor changes to some lists in the tests

This commit is contained in:
Pablito 2024-12-04 14:44:57 +01:00
parent dddc326d0d
commit 20c3200a62

View File

@ -35,21 +35,14 @@ def test_legendre_1():
def test_legendre_2(): def test_legendre_2():
"""Test the 2nd Legendre polynomial.""" """Test the 2nd Legendre polynomial."""
assert legendre(2) == [ assert legendre(2) == [-0.5, 0.0, 1.5]
-0.5, "The 2nd Legendre polynomial should be [-0.5, 0.0, 1.5]"
0.0,
1.5,
], "The 2nd Legendre polynomial should be [-0.5, 0.0, 1.5]"
def test_legendre_3(): def test_legendre_3():
"""Test the 3rd Legendre polynomial.""" """Test the 3rd Legendre polynomial."""
assert legendre(3) == [ assert legendre(3) == [0.0, -1.5, 0.0, 2.5]
0.0, "The 3rd Legendre polynomial should be [0.0, -1.5, 0.0, 2.5]"
-1.5,
0.0,
2.5,
], "The 3rd Legendre polynomial should be [0.0, -1.5, 0.0, 2.5]"
def test_legendre_4(): def test_legendre_4():