Suppress ComplexWarning

ratt-ru · Jan 30, 2024 · 5923cc7 · 5923cc7
1 parent 2945c19
commit 5923cc7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/daskms/tests/test_dataset.py b/daskms/tests/test_dataset.py
@@ -3,6 +3,7 @@
 from itertools import product
 import os
 import uuid
+import warnings
 
 import dask
 import dask.array as da
@@ -305,7 +306,9 @@ def test_dataset_add_column(ms, dtype):
     ds = datasets[0]
 
     # Create the dask array
-    bitflag = da.zeros_like(ds.DATA.data, dtype=dtype)
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore", np.exceptions.ComplexWarning)
+        bitflag = da.zeros_like(ds.DATA.data, dtype=dtype)
     # Assign keyword attribute
     col_kw = {
         "BITFLAG": {