def test_detect_banded_matrix():
m1, m2 = detect_banded_matrix(BANDED_MATRIX)
assert (m1, m2) == (2, 1)
m1, m2 = detect_banded_matrix(TRICKY, check_all=False)
assert (m1, m2) == (2, 1)
m1, m2 = detect_banded_matrix(TRICKY, check_all=True)
assert (m1, m2) == (5, 6)
assert detect_banded_matrix(Matrix(shape=(10, 10))) == (0, 0)
identity = Matrix.identity(shape=(10, 10))
assert detect_banded_matrix(identity) == (0, 0)
def test_identity():
m = Matrix.identity((3, 4))
for i in range(3):
assert m[i, i] == 1.0
