From 480eab9011676d76e04af2a23b6eba2a552c0895 Mon Sep 17 00:00:00 2001 From: Han Wang Date: Sat, 27 Jan 2024 23:25:39 +0800 Subject: [PATCH] fix ut --- source/tests/test_model_format_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/source/tests/test_model_format_utils.py b/source/tests/test_model_format_utils.py index 22393515ec..da76c53ed9 100644 --- a/source/tests/test_model_format_utils.py +++ b/source/tests/test_model_format_utils.py @@ -367,4 +367,5 @@ def test_self_consistency( em1 = DescrptSeA.deserialize(em0.serialize()) mm0 = em0.call(self.coord_ext, self.atype_ext, self.nlist) mm1 = em1.call(self.coord_ext, self.atype_ext, self.nlist) - np.testing.assert_allclose(mm0, mm1) + for ii in [0, 1, 4]: + np.testing.assert_allclose(mm0[ii], mm1[ii])