From d7ecbbaad4ef27c58a2e2eec9e48a6987fedb0e9 Mon Sep 17 00:00:00 2001 From: anyangml Date: Thu, 14 Mar 2024 10:01:27 +0000 Subject: [PATCH] fix: UTs --- source/tests/tf/test_fitting_dos.py | 17 ++++------------- 1 file changed, 4 insertions(+), 13 deletions(-) diff --git a/source/tests/tf/test_fitting_dos.py b/source/tests/tf/test_fitting_dos.py index 353926dc10..bcf3b8368e 100644 --- a/source/tests/tf/test_fitting_dos.py +++ b/source/tests/tf/test_fitting_dos.py @@ -189,22 +189,13 @@ def test_fitting(self): pred_atom_dos = pred_atom_dos.reshape(-1, numb_dos) ref_atom_dos_1 = [ - -0.32495014, - -0.87979356, - -0.26630668, - -0.32495882, - -0.87979767, - -0.2663072, + -0.32495014, -0.32495882, -0.32496842, -0.32495892, -0.32495469, + -0.32496075 ] ref_atom_dos_2 = [ - -0.26630917, - 0.21549911, - -0.87979638, - -0.26630564, - 0.21550413, - -0.87979585, + 0.21549911, 0.21550413, 0.21551077, 0.21550547, 0.21550303, + 0.21550645 ] places = 4 - np.testing.assert_almost_equal(pred_atom_dos[:, 0], ref_atom_dos_1, places) np.testing.assert_almost_equal(pred_atom_dos[:, 50], ref_atom_dos_2, places)