def test_ex3():
"""
Simple 2x2 example with unique solution
:return:
"""
X = Matrix([[7, 4], [-2, 5]])
b = Matrix([[-5], [26]])
A, b_solved = solve_gauss(X, b)
assert A == eye(2) and b_solved.round(2) == Matrix([[-3], [4]])
def test_ex4():
"""
Simple 2x2 example with unique solution
:return:
"""
X = Matrix([[2, 4], [3, 5]])
b = Matrix([[-12], [-16]])
A, b_solved = solve_gauss(X, b)
assert A == eye(2) and b_solved.round(2) == Matrix([[-2, -2]]).T
def test_ex1():
"""
Simple 2x2 example with unique solution
:return:
"""
X = Matrix([[3, -1], [-4, 2]])
b = Matrix([[4], [2]])
A, b_solved = solve_gauss(X, b)
assert A == eye(2) and b_solved.round(2) == Matrix([[5], [11]])
def test_ex2():
"""
Simple 2x2 example with unique solution
:return:
"""
X = Matrix([[3, 4], [-6, 3]])
b = Matrix([[10], [-9]])
A, b_solved = solve_gauss(X, b)
assert A == eye(2) and b_solved.round(2) == Matrix([[2], [1]])
def test_non_solvable():
"""
Simple 2x2 example with no solution
:return:
"""
X = Matrix([[7, 1], [14, 2]])
b = Matrix([[-3], [1]])
A, b_solved = solve_gauss(X, b)
assert A != eye(2)
def test_too_many_variables():
"""
3 equation, but 4 variables -> infinite number of solutions
:return:
"""
X = Matrix([[0, 2, 1, 0], [1, -2, -3, 5], [-1, 1, 2, 2]])
b = Matrix([[-4, 5, -2]]).T
A, b_solved = solve_gauss(X, b)
assert A != eye(3) and b_solved.round(2) != Matrix([[-4, 5, -2]]).T
def test_too_many_equations():
"""
4 equations, but 3 variables, ie. too many constraints -> no solution
:return:
"""
X = Matrix([[0, 2, 1], [1, -2, -3], [-1, 1, 2], [0, 5, 2]])
b = Matrix([[-8], [0], [3], [2]])
A, b_solved = solve_gauss(X, b)
assert A != eye(3) and b_solved.round(2) != Matrix([[-4, 5, -2]]).T
def test_ex5():
"""
Simple 2x2 example with unique solution
:return:
"""
X = Matrix([[1, 2], [2, 4]])
b = Matrix([[5], [2]])
A, b_solved = solve_gauss(X, b)
assert A != eye(2)
def test_ex6():
"""
Simple examples from textbook
:return:
"""
X = Matrix([[1, 3, 4], [2, 7, 3], [2, 8, 6]])
b = Matrix([[3], [-7], [-4]])
A, b_solved = solve_gauss(X, b)
assert A == eye(3) and b_solved.round(2) == Matrix([[4, -3, 2]]).T
X = Matrix([[2, 8, -4], [2, 11, 5], [4, 18, 3]])
b = Matrix([[0], [9], [11]])
A, b_solved = solve_gauss(X, b)
assert A == eye(3) and b_solved.round(2) == Matrix([[2, 0, 1]]).T
X = Matrix([[0, 2, 6], [3, 9, 4], [1, 3, 5]])
b = Matrix([[2], [7], [6]])
A, b_solved = solve_gauss(X, b)
assert A == eye(3) and b_solved.round(2) == Matrix([[7, -2, 1]]).T
X = Matrix([[1, 3, 2, 5], [-1, 2, -2, 5], [2, 6, 4, 7], [0, 5, 2, 6]])
b = Matrix([[11], [-6], [19], [5]])
A, b_solved = solve_gauss(X, b)
assert A == eye(4) and b_solved.round(2) == Matrix([[5, -1, 2, 1]]).T