kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-04 14:43:08 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: can run tests independently

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2024-10-20 18:39:01 +02:00
parent 838d5ed563
commit 9014c37064
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
code/tests.py
View File

@ -5,7 +5,7 @@ from matrix_generator import MatrixGenerator
from richardson_method import RichardsonMethod
@pytest.mark.parametrize("n", [2, 3, 4, 5, 10, 20, 50, 100])
def test_richardson_vs_cg(n):
def test_richardson_vs_cg(n: int):
tolerance = 1e-5
A, b = MatrixGenerator.generate_random_matrix_and_vector(n)
@ -28,7 +28,11 @@ def assert_scipy_converged(solution_richardson, solution_cg, tolerance, A, b):
else:
difference = np.linalg.norm(solution_richardson - solution_cg)
print(f"Difference between Richardson and CG solutions: {difference:.8f}")
if difference >= tolerance:
if difference < tolerance:
print("Both Richardson and CG converged and calculated correct values.")
print("Solution CG:\n", solution_cg)
print("Solution Richardson:\n", solution_richardson)
else:
print("Matrix A:\n", A)
print("Vector b:\n", b)
assert difference < tolerance, f"The solutions are different! Difference: {difference:.8f}"
@ -41,3 +45,8 @@ def assert_scipy_not_converged(solution_richardson, A, b):
print("Matrix A:\n", A)
print("Vector b:\n", b)
assert False, "Richardson converged while SciPy did not"
if __name__ == "__main__":
# Run pytest and exit with the appropriate status code
for n in [2, 3, 4, 5, 10, 20, 50, 100]:
test_richardson_vs_cg(n)