def test_raise_shape_match_error(self):
"""Tests whether an error is thrown when b doesn't match a."""
# B has more rows than a has columns!
a = np.array([[1, 0], [0, 1]])
b = np.array([[4, 1], [2, 2], [2, 2]])
# Check if error is raised
with self.assertRaises(ValueError):
matmul(a, b)
def test_raise_shape_error(self):
"""Tests if error is raised when either A or B does not have 2
dimensions"""
# A does not have 2 dimensions
a = np.array([[[1, 0], [0, 1]], [[0, 1], [0, 1]]])
b = np.array([[4, 1], [2, 2]])
with self.assertRaises(ValueError):
matmul(a, b)
# B does not have 2 dimensions
a = np.array([[1, 0], [0, 1]])
b = np.array([[[1, 0], [0, 1]], [[0, 1], [0, 1]]])
with self.assertRaises(ValueError):
matmul(a, b)
def test_raise_shape_error(self):
"""Tests if error is raised when either A or B does not have 2
dimensions"""
# A does not have 2 dimensions
a = np.array([[self.a1, self.a2], [self.a2, self.a2]])
b = np.array([self.b1, self.b2])
with self.assertRaises(ValueError):
matmul(a, b)
# B does not have 2 dimensions
a = np.array([self.a1, self.a2])
b = np.array([[self.a1, self.a2], [self.a2, self.a2]])
with self.assertRaises(ValueError):
matmul(a, b)
def test_multiplication(self):
"""Test the multiplication itself."""
# Define matrix content.
a = np.array([[1, 0], [0, 1]])
b = np.array([[4, 1], [2, 2]])
# Check result
self.assertTrue(np.all(np.array([[4, 1], [2, 2]] == matmul(a, b))))
def test_multiplication(self):
"""Test the multiplication itself."""
# Define matrix content.
a = np.array([self.a1, self.a2])
b = np.array([self.b1, self.b2])
# Check result
self.assertTrue(np.all(np.array([self.b1, self.b2] == matmul(a, b))))