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

feat: add distributed arrays to processing type

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2024-12-05 20:24:53 +01:00
parent 3721418264
commit afd9e90a38
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
code/tests.py
View File

@ -26,9 +26,30 @@ def calcualte_norm_from_matrix_numpy(A, n):
def solution_lib(A, b): def solution_lib(A, b):
return np.linalg.solve(A, b) return np.linalg.solve(A, b)
@pytest.mark.parametrize("n", [2, 5, 10, 50, 100, 300, 500, 750, 1000, 5000, 10000]) @pytest.mark.parametrize("n", [
@pytest.mark.parametrize("processing_type", [ProcessingType.SEQUENTIAL, ProcessingType.THREADS, ProcessingType.PROCESSES]) 2,
@pytest.mark.parametrize("matrix_type", ["spd", "nemeth12", "poli3"]) 5,
10,
50,
100,
300,
500,
750,
1000,
5000,
10000
])
@pytest.mark.parametrize("processing_type", [
ProcessingType.SEQUENTIAL,
ProcessingType.THREADS,
ProcessingType.PROCESSES,
ProcessingType.DISTRIBUTED_ARRAYS
])
@pytest.mark.parametrize("matrix_type", [
"spd",
"nemeth12",
"poli3"
])
def test_richardson_vs_cg(n: int, processing_type: ProcessingType, matrix_type: str, capsys): def test_richardson_vs_cg(n: int, processing_type: ProcessingType, matrix_type: str, capsys):
print("matrix type: ", matrix_type) print("matrix type: ", matrix_type)
print("matrix size: ", n if matrix_type == "spd" else "fixed") print("matrix size: ", n if matrix_type == "spd" else "fixed")