Update interop.py (arrayfire#103)

ZacDiggum · pavanky · commit cafc208b9b0d · 2016-10-11T10:27:47.000-07:00
* Update interop.py

add numpy u16 dtype

* Add files via upload

* added numpy u16 and s16 dtypes
diff --git a/arrayfire/interop.py b/arrayfire/interop.py
@@ -27,18 +27,19 @@
 
     AF_NUMPY_FOUND=True
 
-    _nptype_to_aftype = {'f4' : Dtype.f32,
+    _nptype_to_aftype = {'b1' : Dtype.b8,
+			 'u1' : Dtype.u8,
+			 'u2' : Dtype.u16,
+			 'i2' : Dtype.s16,
+			 's4' : Dtype.u32,
+			 'i4' : Dtype.s32,
+			 'f4' : Dtype.f32,
+			 'c8' : Dtype.c32,
+			 's8' : Dtype.u64,
+			 'i8' : Dtype.s64,
                          'f8' : Dtype.f64,
-                         'b1' : Dtype.b8,
-                         'u1' : Dtype.u8,
-                         'i4' : Dtype.s32,
-                         's4' : Dtype.u32,
-                         'i8' : Dtype.s64,
-                         's8' : Dtype.u64,
-                         'c8' : Dtype.c32,
                          'c16' : Dtype.c64}
 
-
     def np_to_af_array(np_arr):
         """
         Convert numpy.ndarray to arrayfire.Array.
