def test_matrix_check_dims_match():
a = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
b = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
assert a._check_dims_match(b)
assert not a._check_dims_match(t(a))
def test_matrix_setitem():
"""Test matrix element setter """
a = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
a[0, 0] = 4
assert a[0, 0] == 4
def test_matrix_getitem():
"""Test element getter"""
a = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
assert a[0, 0] == 1
def test_matrix_from_matrix():
"""Test matrix matrix constructor"""
a = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
b = Matrix.from_matrix(a)
assert b.get_data() == [[1, 2, 3], [4, 5, 6]]
def test_matrix_from_list():
"""Test matrix list constructor"""
a = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
assert a.get_data() == [[1, 2, 3], [4, 5, 6]]
def test_t():
"""Test matrix transpose"""
a = Matrix.from_list([[1, 2, 3], [4, 5, 6]])
assert t(a).get_data() == [[1, 4], [2, 5], [3, 6]]