add UT for pt2tf model.deserialize

Author: ChiahsinChu

source/tests/consistent/model/test_dipole.py

@@ -204,3 +204,11 @@ def extract_ret(self, ret: Any, backend) -> tuple[np.ndarray, ...]:
                 ret[1].ravel(),
             )
         raise ValueError(f"Unknown backend: {backend}")
+
+    def test_atom_exclude_types(self):
+        _ret, data = self.get_reference_ret_serialization(self.RefBackend.PT)
+        data["atom_exclude_types"] = [1]
+        self.reset_unique_id()
+        tf_obj = self.tf_class.deserialize(data, suffix=self.unique_id)
+        pt_obj = self.pt_class.deserialize(data)
+        self.assertEqual(tf_obj.get_sel_type(), pt_obj.get_sel_type())

source/tests/consistent/model/test_polar.py

@@ -198,3 +198,11 @@ def extract_ret(self, ret: Any, backend) -> tuple[np.ndarray, ...]:
                 ret[1].ravel(),
             )
         raise ValueError(f"Unknown backend: {backend}")
+
+    def test_atom_exclude_types(self):
+        _ret, data = self.get_reference_ret_serialization(self.RefBackend.PT)
+        data["atom_exclude_types"] = [1]
+        self.reset_unique_id()
+        tf_obj = self.tf_class.deserialize(data, suffix=self.unique_id)
+        pt_obj = self.pt_class.deserialize(data)
+        self.assertEqual(tf_obj.get_sel_type(), pt_obj.get_sel_type())