

def test_02_upper(self):
    ab = array([[0.0, 0.0, 2.0, 2.0], [(- 99), 1.0, 1.0, 1.0], [4.0, 4.0, 4.0, 4.0]])
    b = array([[1.0, 6.0], [4.0, 2.0], [1.0, 6.0], [2.0, 1.0]])
    x = solveh_banded(ab, b)
    expected = array([[0.0, 1.0], [1.0, 0.0], [0.0, 1.0], [0.0, 0.0]])
    assert_array_almost_equal(x, expected)

dataset/ETHPy150Open scipy/scipy/scipy/linalg/tests/test_basic.py TestSolveHBanded.test_02_upper/original