def test_output_4(N):
N = randint(1, 20)
A = np.random.rand(N, N)
x = np.ones((N, 1))
b = np.dot(A, x)
assert pytest.approx(sm(A, b), x)
def test_output_array():
A = np.ones((1, 1))
b = np.ones(1)
assert isinstance(sm(A, b), np.ndarray), \
"the output temperature should be an array."
def test_output_2():
A = np.array([[1, 1], [15, 50]])
b = np.array([[6], [20]])
assert sm(A, b).shape == b.shape
def test_output_3():
A = np.array([[5, -6, -7], [6, -4, 10], [2, 4, -3]])
b = np.array([[7], [-34], [29]])
assert pytest.approx(sm(A, b), np.array([[2], [4], [-3]]))
def test_output_1():
A = np.ones((1, 1))
b = np.ones(1)
assert sm(A, b) == 1