def test_row_col():
m = DenseMatrix(3, 3, [1, 2, 3, 4, 5, 6, 7, 8, 9])
assert m.row(0) == DenseMatrix(1, 3, [1, 2, 3])
assert m.col(0) == DenseMatrix(3, 1, [1, 4, 7])
def test_row_col():
m = DenseMatrix(3, 3, [1, 2, 3, 4, 5, 6, 7, 8, 9])
assert m.row(0) == DenseMatrix(1, 3, [1, 2, 3])
assert m.col(0) == DenseMatrix(3, 1, [1, 4, 7])