fix: UT

Author: anyangml

tests/test_deepmd_mixed.py

@@ -5,6 +5,11 @@
 import unittest
 from glob import glob
 
+from dpdata.data_type import (
+    Axis,
+    DataType,
+)
+
 import numpy as np
 from comp_sys import (
     CompLabeledMultiSys,
@@ -464,6 +469,27 @@ def setUp(self):
         self.f_places = 6
         self.v_places = 6
 
+        new_datatypes=[
+
+                DataType(
+                    "fparam",
+                    np.ndarray,
+                    shape=(Axis.NFRAMES, 2),
+                    required=False,
+                ),
+                DataType(
+                    "aparam",
+                    np.ndarray,
+                    shape=(Axis.NFRAMES, Axis.NATOMS, 2),
+                    required=False,
+                ),
+            ]
+            
+        for datatype in new_datatypes:
+            dpdata.System.register_data_type(datatype)
+            dpdata.LabeledSystem.register_data_type(datatype)
+
+
         # C1H4
         system_1 = dpdata.LabeledSystem(
             "gaussian/methane.gaussianlog", fmt="gaussian/log"