Change assert_equal to assert_almost_equal

Chengqian-Zhang · Chengqian-Zhang · commit 1bba83314750 · 2025-08-05T10:50:28.000Z
diff --git a/source/tests/pt/test_dp_test.py b/source/tests/pt/test_dp_test.py
@@ -203,7 +203,7 @@ def test_dp_test_padding_atoms(self) -> None:
                 input_dict_padding["coord"], (0, 0, 0, padding_atoms, 0, 0), value=0
             )
             result_padding = trainer.model(**input_dict_padding)
-            np.testing.assert_equal(
+            np.testing.assert_almost_equal(
                 to_numpy_array(result[trainer.model.get_var_name()])[0],
                 to_numpy_array(result_padding[trainer.model.get_var_name()])[0],
             )

Original file line number	Diff line number	Diff line change
`@@ -203,7 +203,7 @@ def test_dp_test_padding_atoms(self) -> None:`
`203`	`203`	`input_dict_padding["coord"], (0, 0, 0, padding_atoms, 0, 0), value=0`
`204`	`204`	`)`
`205`	`205`	`result_padding = trainer.model(**input_dict_padding)`
`206`		`- np.testing.assert_equal(`
	`206`	`+ np.testing.assert_almost_equal(`
`207`	`207`	`to_numpy_array(result[trainer.model.get_var_name()])[0],`
`208`	`208`	`to_numpy_array(result_padding[trainer.model.get_var_name()])[0],`
`209`	`209`	`)`