def test_correct_array(self):
arr_a = [
[2, 5, 4],
[1, 3, 2],
[2, 10, 9],
]
arr_b = [30, 150, 110]
self.assertTupleEqual(solve(arr_a, arr_b), (-152, 270, -254))
arr_a = [
[1, -1],
[3, 2],
]
arr_b = [7, 16]
self.assertTupleEqual(solve(arr_a, arr_b), (6, -1))
arr_a = [
[2, -4, 1, -5],
[4, -7, -1, -8],
[10, -18, 2, -23],
[2, -3, 1, -1],
]
arr_b = [2, -5, 3, 0]
self.assertTupleEqual(solve(arr_a, arr_b), (1, 2, 3, -1))
arr_a = [2]
arr_b = [14]
self.assertTupleEqual(solve(arr_a, arr_b), (7, ))
def test_array_with_zero_determinant(self):
arr_a = [
[1, -2],
[-2, 4],
]
arr_b = [6, 0]
with self.assertRaises(ValueError):
solve(arr_a, arr_b)
def test_non_quadratic_array(self):
arr_a = [
[1, 5],
[13, 60],
[14, 0],
]
arr_b = [15, 17]
with self.assertRaises(ValueError):
solve(arr_a, arr_b)
arr_a = [4, 6, 8]
arr_b = [6, 5, 0]
with self.assertRaises(ValueError):
solve(arr_a, arr_b)
def test_wrong_size_array(self):
arr_a = [2]
arr_b = [14, 4]
with self.assertRaises(ValueError):
solve(arr_a, arr_b)
arr_a = [
[1, 4],
[5, 10],
]
arr_b = [6, 5, 0]
with self.assertRaises(ValueError):
solve(arr_a, arr_b)
arr_a = [
[1, 4],
[5, 10],
[4, 6, 8],
]
arr_b = [6, 5, 0]
with self.assertRaises(ValueError):
solve(arr_a, arr_b)
def test_empty_array(self):
with self.assertRaises(ValueError):
solve([], [])