xxxxx

njzjz · Aug 28, 2024 · 799f3b8 · 799f3b8
1 parent b1fec4e
commit 799f3b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_custom_data_type.py b/tests/test_custom_data_type.py
@@ -48,7 +48,7 @@ def test_from_deepmd_hdf5(self):
         np.testing.assert_allclose(x.data["foo"], self.foo)
 
     def test_duplicated_data_type(self):
-        dt = DataType("foo", np.ndarray, (Axis.NFRAMES, 2, 4), required=False)
+        dt = DataType("foo", np.ndarray, (Axis.NFRAMES, *self.shape), required=False)
         n_dtypes_old = len(self.cls.DTYPES)
         with self.assertWarns(UserWarning):
             self.cls.register_data_type(dt)