add test

legend-exp · Dec 7, 2024 · e40e468 · e40e468
1 parent 530206b
commit e40e468
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_lgdo_utils.py b/tests/test_lgdo_utils.py
@@ -1,11 +1,15 @@
 from __future__ import annotations
 
+import h5py
 import numpy as np
 
 from lgdo import utils
 
 
 def test_get_element_type():
+    # variable length HD5 string datatype.
+    h5py_str_dtype = h5py.string_dtype(encoding="ascii", length=None)
+
     objs = [
         ("hi", "string"),
         (True, "bool"),
@@ -16,6 +20,7 @@ def test_get_element_type():
         (1 + 1j, "complex"),
         (b"hi", "string"),
         (np.array(["hi"]), "string"),
+        (np.array([b"hi"], h5py_str_dtype), "string"),
     ]
 
     for obj, name in objs: