<reponame>avishekrk/DFI
# IPython log file


def test_hessian():

    import dfi
    import numpy as np

    a = np.array([64, 0, 0, -64, 0, 0], dtype=float)
    b = np.array([0, 0, 0, 0, 0, 0], dtype=float)
    c = np.array([-64, 0, 0, 64, 0, 0], dtype=float)

    test_hess = np.vstack((a, b, b, c, b, b))

    x = np.array([0, 5], dtype=float)
    y = np.array([0, 0], dtype=float)
    z = np.array([0, 0], dtype=float)
    resnum = 2
    gamma = 100
    hess = dfi.dfi_calc.calchessian(resnum, x, y, z, gamma)
    assert np.all(test_hess == hess)

# IPython log file


def test_hessian():

    import dfi
    import numpy as np

    a = np.array([64, 0, 0, -64, 0, 0], dtype=float)
    b = np.array([0, 0, 0, 0, 0, 0], dtype=float)
    c = np.array([-64, 0, 0, 64, 0, 0], dtype=float)

    test_hess = np.vstack((a, b, b, c, b, b))

    x = np.array([0, 5], dtype=float)
    y = np.array([0, 0], dtype=float)
    z = np.array([0, 0], dtype=float)
    resnum = 2
    gamma = 100
    hess = dfi.dfi_calc.calchessian(resnum, x, y, z, gamma)
    assert np.all(test_hess == hess)
# IPython log file