Fix checks whether alloc arguments are omitted

European-XFEL · Mar 29, 2021 · e729be7 · e729be7
1 parent 8cf52e3
commit e729be7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pasha/context.py b/pasha/context.py
@@ -51,8 +51,8 @@ def _resolve_alloc(shape, dtype, order, like):
                 memory order.
         """
         if like is not None:
-            shape = shape or like.shape
-            dtype = dtype or like.dtype
+            shape = shape if shape is not None else like.shape
+            dtype = dtype if dtype is not None else like.dtype
 
             try:
                 if order is None:
@@ -76,8 +76,8 @@ def _resolve_alloc(shape, dtype, order, like):
             raise ValueError('array shape must be specified')
 
         else:
-            dtype = dtype or np.float64
-            order = order or 'C'
+            dtype = dtype if dtype is not None else np.float64
+            order = order if order is not None else 'C'
 
         if isinstance(shape, int):
             shape = (shape,)

diff --git a/pasha/tests/test_context.py b/pasha/tests/test_context.py
@@ -14,7 +14,6 @@
 from pasha.functor import Functor
 
 
-
 @pytest.mark.parametrize(
     'ctx_cls', [MapContext, ProcessContext],
     ids=['MapContext', 'ProcessContext'])