Merge pull request #143 from mhmerrill/pdarray-index-enhancement

Pdarray index enhancement

.gitignore

@@ -2,6 +2,7 @@ arkouda_server
 arkouda_server_real
 arkouda_server_llvm
 arkouda_server_llvm_real
+*.log
 *_real
 #*#
 .#*

check.py

@@ -8,17 +8,190 @@
 
 import arkouda as ak
 
+print(">>> Sanity checks on the arkouda_server")
+
 ak.v = False
 if len(sys.argv) > 1:
     ak.connect(server=sys.argv[1], port=sys.argv[2])
 else:
     ak.connect()
 
-a = ak.array(np.arange(0,10,1))
-b = ak.arange(0,10,1)
-print(a,b)
-c = a == b
-print(type(c),c)
-print(c.all())
 
+N = 1_000_000
+
+
+def pass_fail(f):
+    return ("Passed" if f else "Failed")
+
+def check_arange(N):
+    # create np version
+    a = ak.array(np.arange(N))
+    # create ak version
+    b = ak.arange(N)
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check arange :", check_arange(N))
+
+def check_linspace(N):
+    # create np version
+    a = ak.array(np.linspace(10, 20, N))
+    # create ak version
+    b = ak.linspace(10, 20, N)
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check linspace :", check_linspace(N))
+
+def check_ones(N):
+    # create np version
+    a = ak.array(np.ones(N))
+    # create ak version
+    b = ak.ones(N)
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check ones :", check_ones(N))
+
+def check_zeros(N):
+    # create np version
+    a = ak.array(np.zeros(10))
+    # create ak version
+    b = ak.zeros(10)
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check zeros :", check_zeros(N))
+
+def check_get_slice(N):
+    # create np version
+    a = np.ones(N)
+    a = a[::2]
+    a = ak.array(a)
+    # create ak version
+    b = ak.ones(N)
+    b = b[::2]
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check get slice [::2] :", check_get_slice(N))
+
+def check_set_slice(N):
+    # create np version
+    a = np.ones(N)
+    a[::2] = 0
+    a = ak.array(a)
+    # create ak version
+    b = ak.ones(N)
+    b[::2] = 0
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check set slice [::2] :", check_set_slice(N))
+
+def check_get_bool_iv(N):
+    # create np version
+    a = np.arange(N)
+    a = a[a < N//2]
+    a = ak.array(a)
+    # create ak version
+    b = ak.arange(N)
+    b = b[b < N//2]
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check (compressing) get bool iv :", check_get_bool_iv(N))
+
+def check_get_integer_iv(N):
+    # create np version
+    a = np.arange(N)
+    iv = np.arange(N//2)
+    a = a[iv]
+    a = ak.array(a)
+    # create ak version
+    b = ak.arange(N)
+    iv = ak.arange(N//2)
+    b = b[iv]
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check (gather) get integer iv :", check_get_integer_iv(N))
+
+def check_set_integer_iv(N):
+    # create np version
+    a = np.arange(N)
+    iv = np.arange(N//2)
+    a[iv] = iv*10
+    a = ak.array(a)
+    # create ak version
+    b = ak.arange(N)
+    iv = ak.arange(N//2)
+    b[iv] = iv*10
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check (scatter) set integer iv :", check_set_integer_iv(N))
+
+def check_get_integer_idx(N):
+    # create np version
+    a = np.arange(N)
+    v1 = a[N//2]
+    # create ak version
+    b = ak.arange(N)
+    v2 = b[N//2]
+    return pass_fail(v1 == v2)
+
+print("check get integer idx :", check_get_integer_idx(N))
+
+def check_set_integer_idx(N):
+    # create np version
+    a = np.arange(N)
+    a[N//2] = -1
+    v1 = a[N//2]
+    # create ak version
+    b = ak.arange(N)
+    b[N//2] = -1
+    v2 = b[N//2]
+    return pass_fail(v1 == v2)
+
+print("check set integer idx :", check_set_integer_idx(N))
+
+def check_argsort(N):
+    # create np version
+    a = np.arange(N)
+    a = a[::-1]
+    iv = np.argsort(a)
+    a = a[iv]
+    a = ak.array(a)
+    # create ak version
+    b = ak.arange(N)
+    b = b[::-1]
+    iv = ak.argsort(b)
+    b = b[iv]
+    # print(a,b)
+    c = a == b
+    # print(type(c),c)
+    return pass_fail(c.all())
+
+print("check argsort :", check_argsort(N))
+
+#ak.disconnect()
 ak.shutdown()