Adding tests for array class

- Also changed checks for data types in data creation

Author: pavanky

arrayfire/array.py

@@ -11,8 +11,12 @@ def create_array(buf, numdims, idims, dtype):
     return out_arr
 
 def constant_array(val, d0, d1=None, d2=None, d3=None, dtype=f32):
+
     if not isinstance(dtype, c_int):
-        raise TypeError("Invalid dtype")
+        if isinstance(dtype, int):
+            dtype = c_int(dtype)
+        else:
+            raise TypeError("Invalid dtype")
 
     out = c_longlong(0)
     dims = dim4(d0, d1, d2, d3)
@@ -130,7 +134,7 @@ def dims(self):
     def type(self):
         dty = c_int(f32.value)
         safe_call(clib.af_get_type(pointer(dty), self.arr))
-        return dty
+        return dty.value
 
     def __add__(self, other):
         return binary_func(self, other, clib.af_add)

arrayfire/data.py

@@ -12,8 +12,12 @@ def constant(val, d0, d1=None, d2=None, d3=None, dtype=f32):
 brange = range
 
 def range(d0, d1=None, d2=None, d3=None, dim=-1, dtype=f32):
+
     if not isinstance(dtype, c_int):
-        raise TypeError("Invalid dtype")
+        if isinstance(dtype, int):
+            dtype = c_int(dtype)
+        else:
+            raise TypeError("Invalid dtype")
 
     out = array()
     dims = dim4(d0, d1, d2, d3)
@@ -24,7 +28,10 @@ def range(d0, d1=None, d2=None, d3=None, dim=-1, dtype=f32):
 
 def iota(d0, d1=None, d2=None, d3=None, dim=-1, tile_dims=None, dtype=f32):
     if not isinstance(dtype, c_int):
-        raise TypeError("Invalid dtype")
+        if isinstance(dtype, int):
+            dtype = c_int(dtype)
+        else:
+            raise TypeError("Invalid dtype")
 
     out = array()
     dims = dim4(d0, d1, d2, d3)
@@ -42,7 +49,10 @@ def iota(d0, d1=None, d2=None, d3=None, dim=-1, tile_dims=None, dtype=f32):
 def randu(d0, d1=None, d2=None, d3=None, dtype=f32):
 
     if not isinstance(dtype, c_int):
-        raise TypeError("Invalid dtype")
+        if isinstance(dtype, int):
+            dtype = c_int(dtype)
+        else:
+            raise TypeError("Invalid dtype")
 
     out = array()
     dims = dim4(d0, d1, d2, d3)
@@ -53,7 +63,10 @@ def randu(d0, d1=None, d2=None, d3=None, dtype=f32):
 def randn(d0, d1=None, d2=None, d3=None, dtype=f32):
 
     if not isinstance(dtype, c_int):
-        raise TypeError("Invalid dtype")
+        if isinstance(dtype, int):
+            dtype = c_int(dtype)
+        else:
+            raise TypeError("Invalid dtype")
 
     out = array()
     dims = dim4(d0, d1, d2, d3)
@@ -72,7 +85,10 @@ def get_seed():
 def identity(d0, d1=None, d2=None, d3=None, dtype=f32):
 
     if not isinstance(dtype, c_int):
-        raise TypeError("Invalid dtype")
+        if isinstance(dtype, int):
+            dtype = c_int(dtype)
+        else:
+            raise TypeError("Invalid dtype")
 
     out = array()
     dims = dim4(d0, d1, d2, d3)

tests/simple_array.py

@@ -0,0 +1,15 @@
+#!/usr/bin/python
+import arrayfire as af
+import array as host
+
+a = af.array([1, 2, 3])
+af.print_array(a)
+print(a.numdims(), a.dims(), a.type())
+
+a = af.array(host.array('d', [4, 5, 6]))
+af.print_array(a)
+print(a.numdims(), a.dims(), a.type())
+
+a = af.array(host.array('l', [7, 8, 9] * 4), (2, 5))
+af.print_array(a)
+print(a.numdims(), a.dims(), a.type())