diff options
Diffstat (limited to 'test/test_solvers.py')
| -rw-r--r-- | test/test_solvers.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/test/test_solvers.py b/test/test_solvers.py index e5b73d8..5b5da6f 100644 --- a/test/test_solvers.py +++ b/test/test_solvers.py @@ -3,8 +3,10 @@ import numpy as np from numpy.typing import NDArray + import solvers + def test_elimination_3() -> None: """Tests elimination with 3 variables.""" aa = np.array([[2.0, 4.0, 4.0], [5.0, 4.0, 2.0], [1.0, 2.0, -1.0]], dtype=np.float_) @@ -30,3 +32,26 @@ def test_lindep_3() -> None: xx_expected = None xx_gauss = solvers.gaussian_eliminate(aa, bb) assert xx_expected == xx_gauss + + +def test_lu_factorization() -> None: + aa = np.array([[2.0, 4.0, 4.0], [1.0, 2.0, -1.0], [5.0, 4.0, 2.0]]) + ll, uu, pp = solvers.lu_factorization(aa) + assert np.allclose(aa, pp.T @ ll @ uu) + + +def test_forward_substitution() -> None: + ll = np.array([[2.0, 0.0, 0.0], [-1.0, 2.0, 0.0], [4.0, 4.0, 2.0]]) + bb = np.array([3.0, 2.0, 1.0]) + xx_result = solvers.forward_substitution(ll, bb)[:, 0] + xx_expected = np.array([3 / 2, 7.0 / 4.0, -6]) + print(xx_result) + assert np.allclose(xx_expected, xx_result) + +def test_back_substitution() -> None: + uu = np.array([[2.0, 4.0, 4.0], [0.0, 2.0, -1.0], [0.0, 0.0, 2.0]]) + bb = np.array([1.0, 2.0, 3.0]) + xx_result = solvers.back_substitution(uu, bb)[:, 0] + xx_expected = np.array([-6, 7.0 / 4.0, 3.0 / 2.0]) + print(xx_result) + assert np.allclose(xx_expected, xx_result) |