kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-11-11 13:27:38 +01: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 from richardson_method import RichardsonMethod
@pytest.mark.parametrize("n", [2, 3, 4, 5, 10, 20, 50, 100]) @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 tolerance = 1e-5
A, b = MatrixGenerator.generate_random_matrix_and_vector(n) 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: else:
difference = np.linalg.norm(solution_richardson - solution_cg) difference = np.linalg.norm(solution_richardson - solution_cg)
print(f"Difference between Richardson and CG solutions: {difference:.8f}") 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("Matrix A:\n", A)
print("Vector b:\n", b) print("Vector b:\n", b)
assert difference < tolerance, f"The solutions are different! Difference: {difference:.8f}" 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("Matrix A:\n", A)
print("Vector b:\n", b) print("Vector b:\n", b)
assert False, "Richardson converged while SciPy did not" 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)