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]]