Merge pull request #326 from abergeron/bool

Add proper conversion to bool.

Author: nouiz

pygpu/gpuarray.pyx

@@ -1657,6 +1657,14 @@ cdef class GpuArray:
         """
         return pygpu_as_ndarray(self)
 
+    def __bool__(self):
+        if self.size == 0:
+            return False
+        elif self.size == 1:
+            return bool(numpy.asarray(self))
+        else:
+            raise ValueError('The truth value of a multi-element array is ambiguous')
+
     def _empty_like_me(self, dtype=None, order='C'):
         """
         _empty_like_me(dtype=None, order='C')

pygpu/tests/test_gpu_ndarray.py

@@ -47,6 +47,11 @@ def test_hash():
     assert exc is not None
 
 
+def test_bool():
+    for data in [numpy.empty((0, 33)), [[1]], [[0]], [], [0], [1], 0, 1]:
+        assert bool(pygpu.asarray(data, context=ctx)) == bool(numpy.asarray(data))
+
+
 def test_transfer():
     for shp in [(), (5,), (6, 7), (4, 8, 9), (1, 8, 9)]:
         for dtype in dtypes_all: