示例#1
0
 def setUp(self):
     self.matrix = Matrix.fromtextfile("../test_files/test.txt")
示例#2
0
from models import Matrix
from pathlib import Path

a_matrix = [
    "zad1_a.txt", "zad2_a.txt", "zad3_a.txt", "zad4_a.txt", "zad5_a.txt"
]
b_matrix = [
    "zad1_b.txt", "zad2_b.txt", "zad3_b.txt", "zad4_b.txt", "zad5_b.txt"
]

for i in range(len(a_matrix)):
    '''
        For all matrices calculate LU, LUP, forward and backward substitution
    '''
    try:
        matrix = Matrix.fromtextfile(Path("test_files/") / a_matrix[i])
        b = Matrix.fromtextfile(Path("test_files/") / b_matrix[i])
    except FileNotFoundError:
        print("Missing file(s) {}, {}...\n".format(a_matrix[i], b_matrix[i]))
        continue
    print("\n\n**********{} + {}**********\n".format(a_matrix[i], b_matrix[i]))

    try:
        print("############LU:############")
        matrix.lu()
        print("A:")
        print(str(matrix))
        matrix.forward_substitution(b)
        print("\n############LU: Forward substitution:####")
        print("Y:")
        print(str(b))