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)
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_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_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_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)
