Update test_imag_self_energy.py

2023-12-25 16:00:38 +09:00 · 2023-12-25 16:00:38 +09:00 · 748b016b29
parent 1529da23d1
commit 748b016b29
2 changed files with 1175 additions and 839 deletions
--- a/test/conductivity/test_kappa_RTA.py
+++ b/test/conductivity/test_kappa_RTA.py
@ -264,9 +264,7 @@ def test_kappa_RTA_si_N_U(si_pbesol):

    # print(np.sum(gN), np.sum(gU))
    np.testing.assert_allclose(
-        np.sum([np.ravel(gN_ref), np.ravel(gU_ref)], axis=0),
-        gN.ravel() + gU.ravel(),
-        atol=1e-2,
+        np.ravel(gN_ref) + np.ravel(gU_ref), gN.ravel() + gU.ravel(), atol=1e-2
    )
    np.testing.assert_allclose(np.ravel(gN_ref), gN.ravel(), atol=1e-3)
    np.testing.assert_allclose(np.ravel(gU_ref), gU.ravel(), atol=1e-3)
--- a/test/phonon3/test_imag_self_energy.py
+++ b/test/phonon3/test_imag_self_energy.py