def test_partial_energy_matrix_dec1(): E_g = np.array([10.0, 0.0]) E_n = np.array([10.0, 0.0]) pem = partial_energy_matrix_mono(E_g, E_n, -1) expected = np.array([[1.0]]) assert_array_equal(pem, expected)
def test_partial_energy_matrix_inc2(): E_g = np.array([0.0, 5.0, 10.0]) E_n = np.array([0.0, 5.0, 10.0]) pem = partial_energy_matrix_mono(E_g, E_n, 1) expected = np.array([[1.0, 0.0], [0.0, 1.0]]) assert_array_equal(pem, expected)
def test_partial_energy_matrix_inc6(): 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 = partial_energy_matrix_mono(E_g, E_n, 1) expected = np.array([[1.0, 0.6, 0.0, 0.0], [0.0, 0.4, 1.0, 0.2]]) assert_array_equal(pem, expected)
def test_partial_energy_matrix_dec6(): E_g = np.array([8.0, 4.0, 0.0]) E_n = np.array([10.0, 7.5, 5.0, 2.5, 0.0]) pem = partial_energy_matrix_mono(E_g, E_n, -1) expected = np.array([[0.2, 1.0, 0.4, 0.0], [0.0, 0.0, 0.6, 1.0]]) assert_array_equal(pem, expected)