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

chore: moved debug prints to separate function

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2024-10-20 19:20:38 +02:00
parent 70c0da16c0
commit cb179a76ce
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
code/tests.py
View File

@ -23,19 +23,20 @@ def calculate_eigenvalues(A):
return lambda_min, lambda_max return lambda_min, lambda_max
@pytest.mark.parametrize("n", [2, 3, 4, 5, 10, 20, 50, 100]) def numpy_additional_richardson_calculations():
def test_richardson_vs_cg(n: int):
print("matrix size: ", n)
tolerance = 1e-5
max_iterations=1000
A, b = MatrixGenerator.generate_random_matrix_and_vector(n)
lambda_min, lambda_max = calculate_eigenvalues(A) lambda_min, lambda_max = calculate_eigenvalues(A)
print("eigenvalues: ", lambda_min, lambda_max, RichardsonMethod.calculate_eigenvalues(A, max_iterations)) print("eigenvalues: ", lambda_min, lambda_max, RichardsonMethod.calculate_eigenvalues(A, max_iterations))
omega = 2 / (lambda_min + lambda_max) omega = 2 / (lambda_min + lambda_max)
print("omega: ", omega, RichardsonMethod.calculate_omega(lambda_min, lambda_max)) print("omega: ", omega, RichardsonMethod.calculate_omega(lambda_min, lambda_max))
I = np.eye(n) I = np.eye(n)
print("norms: ", calculate_norm_numpy(I, omega, A), RichardsonMethod.convergence_norm(A, omega, I)) print("norms: ", calculate_norm_numpy(I, omega, A), RichardsonMethod.convergence_norm(A, omega, I))
@pytest.mark.parametrize("n", [2, 3, 4, 5, 10, 20, 50, 100])
def test_richardson_vs_cg(n: int):
print("matrix size: ", n)
tolerance = 1e-5
max_iterations=1000
A, b = MatrixGenerator.generate_random_matrix_and_vector(n)
richardson_solver = RichardsonMethod(A, b, max_iterations, size=n, tol=1e-7) richardson_solver = RichardsonMethod(A, b, max_iterations, size=n, tol=1e-7)
solution_richardson = richardson_solver.solve() solution_richardson = richardson_solver.solve()