add n5py complex tests

constantinpape · Oct 28, 2024 · c04d211 · c04d211
1 parent 12f4d0d
commit c04d211
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 7 deletions.
diff --git a/include/z5/types/types.hxx b/include/z5/types/types.hxx
@@ -57,14 +57,16 @@ namespace types {
         static DtypeMap & n5ToDtype() {
             static DtypeMap dtypeMap({{{"int8", int8},  {"int16", int16},  {"int32", int32},  {"int64", int64},
                                        {"uint8", uint8}, {"uint16", uint16}, {"uint32", uint32}, {"uint64", uint64},
-                                       {"float32", float32}, {"float64", float64}}});
+                                       {"float32", float32}, {"float64", float64},
+                                       {"complex64", complex64}, {"complex128", complex128}, {"complex256", complex256}}});
             return dtypeMap;
         }
 
         static InverseDtypeMap & dtypeToN5() {
             static InverseDtypeMap dtypeMap({{{int8   ,"int8"},    {int16,   "int16"},  {int32, "int32"},  {int64, "int64"},
                                               {uint8  ,"uint8"},   {uint16,  "uint16"}, {uint32,"uint32"}, {uint64, "uint64"},
-                                              {float32,"float32"}, {float64, "float64"}}});
+                                              {float32,"float32"}, {float64, "float64"},
+                                              {complex64,"complex64"}, {complex128, "complex128"}, {complex256, "complex256"}}});
             return dtypeMap;
         }
     };

diff --git a/src/python/module/z5py/dataset.py b/src/python/module/z5py/dataset.py
@@ -31,7 +31,10 @@ class Dataset:
                    np.dtype('int32'): 'int32',
                    np.dtype('int64'): 'int64',
                    np.dtype('float32'): 'float32',
-                   np.dtype('float64'): 'float64'}
+                   np.dtype('float64'): 'float64',
+                   np.dtype('complex64'): 'complex64',
+                   np.dtype('complex128'): 'complex128',
+                   np.dtype('complex256'): 'complex256'}
 
     # Compression libraries supported by zarr format
     compressors_zarr = tuple(comp for comp in COMPRESSORS_ZARR if AVAILABLE_COMPRESSORS[comp])

diff --git a/src/python/test/test_compression.py b/src/python/test/test_compression.py
@@ -15,7 +15,8 @@ def setUp(self):
         self.dtypes = [
             'int8', 'int16', 'int32', 'int64',
             'uint8', 'uint16', 'uint32', 'uint64',
-            'float32', 'float64'
+            'float32', 'float64',  # 'float128',
+            'complex64', 'complex128',  # 'complex256',
         ]
 
     def tearDown(self):

diff --git a/src/python/test/test_dataset.py b/src/python/test/test_dataset.py
@@ -16,20 +16,23 @@ def setUp(self):
         self.base_dtypes = [
             'int8', 'int16', 'int32', 'int64',
             'uint8', 'uint16', 'uint32', 'uint64',
-            'float32', 'float64'
+            'float32', 'float64',  # 'float128',
+            'complex64', 'complex128',  # 'complex256',
         ]
         self.dtypes = tuple(
             self.base_dtypes +
             [np.dtype(s) for s in self.base_dtypes] +
             [
                 '<i1', '<i2', '<i4', '<i8',
                 '<u1', '<u2', '<u4', '<u8',
-                '<f4', '<f8'
+                '<f4', '<f8',  # '<f16',
+                '<c8', '<c16',  # '<c32',
             ] +
             [
                 np.int8, np.int16, np.int32, np.int64,
                 np.uint8, np.uint16, np.uint32, np.uint64,
-                np.float32, np.float64
+                np.float32, np.float64,  # np.complex128,
+                np.complex64, np.complex128,  # np.complex256,
             ]
         )