def test_partial_energy_matrix1(): xs.xs_cache.clear() E_g = np.array([0.0, 10.0]) E_n = np.array([0.0, 10.0]) pem = xs.partial_energy_matrix(E_g, E_n) expected = np.array([[1.0]]) assert_array_equal(pem, expected)
def test_partial_energy_matrix6(): xs.xs_cache.clear() E_g = np.array([0.0, 4.0, 8.0]) E_n = np.array([0.0, 2.5, 5.0, 7.5, 10.0]) pem = xs.partial_energy_matrix(E_g, E_n) expected = np.array([[1.0, 0.6, 0.0, 0.0], [0.0, 0.4, 1.0, 0.2]]) assert_array_equal(pem, expected)