testing rands

test_scripts/test_randomplot1.py

@@ -0,0 +1,140 @@
+#!/usr/bin/python3
+
+import os,sys,math
+import numpy as np
+import matplotlib.pyplot as plt
+
+#################
+## Subroutines ##
+#################
+
+def binload_float_ndarray(fp):
+    arr = np.zeros((0),dtype=np.float32,order='F')
+
+    qb = fp.read(4)
+    Nd = np.frombuffer(qb,dtype=np.int32,count=1)[0]
+    shp = np.zeros((Nd),dtype=np.int32)
+
+    piprod = 1
+    for I in range(0,Nd):
+        qb = fp.read(4)
+        shp[I] = np.frombuffer(qb,dtype=np.int32,count=1)[0]
+        piprod = piprod*shp[I]
+    
+    qb = fp.read(4*piprod)
+    arr = np.frombuffer(qb,dtype=np.float32,count=piprod)
+
+    arr = arr.reshape(shp)
+    
+    return arr;
+
+def binload_int_ndarray(fp):
+    arr = np.zeros((0),dtype=np.float32,order='F')
+
+    qb = fp.read(4)
+    Nd = np.frombuffer(qb,dtype=np.int32,count=1)[0]
+    shp = np.zeros((Nd),dtype=np.int32)
+
+    piprod = 1
+    for I in range(0,Nd):
+        qb = fp.read(4)
+        shp[I] = np.frombuffer(qb,dtype=np.int32,count=1)[0]
+        piprod = piprod*shp[I]
+    
+    qb = fp.read(4*piprod)
+    arr = np.frombuffer(qb,dtype=np.int32,count=piprod)
+
+    arr = arr.reshape(shp)
+    
+    return arr;
+
+
+#################
+## Main Script ##
+#################
+
+def periodcheck(arr):
+
+    arr = np.asarray(arr).copy().flatten()
+    N = arr.shape[0]
+    q1 = arr[0]
+    q2 = arr[1]
+    q3 = arr[2]
+    ind = -1
+    for I in range(3,N-3):
+        if(arr[I]==q1):
+            if(arr[I+1]==q2):
+                if(arr[I+2]==q3):
+                    ind = I
+                    break
+    if(ind>=0):
+        print("array has detected a period of {} out of {}".format(ind,N))
+    
+    return ind
+
+def test_1():
+
+    fname = "./test_scripts/randf_array.bin"
+    try:
+        fp = open(fname,"rb")
+    except:
+        print("Could not open {} for reading".format(fname))
+        return
+    arr = binload_float_ndarray(fp)
+    periodcheck(arr)
+    fp.close()
+
+    fname = "./test_scripts/randf_array2.bin"
+    try:
+        fp = open(fname,"rb")
+    except:
+        print("Could not open {} for reading".format(fname))
+        return
+    arr2 = binload_float_ndarray(fp)
+    periodcheck(arr2)
+    fp.close()
+
+    fname = "./test_scripts/randint_array.bin"
+    try:
+        fp = open(fname,"rb")
+    except:
+        print("Could not open {} for reading".format(fname))
+        return
+    arr3 = binload_int_ndarray(fp)
+    periodcheck(arr3)
+    print("array 3 max {} min {}".format(np.max(arr3),np.min(arr3)))
+    fp.close()
+
+    fname = "./test_scripts/randint_array2.bin"
+    try:
+        fp = open(fname,"rb")
+    except:
+        print("Could not open {} for reading".format(fname))
+        return
+    arr4 = binload_int_ndarray(fp)
+    periodcheck(arr4)
+    print("array 4 max {} min {}".format(np.max(arr4),np.min(arr4)))
+    fp.close()
+
+    plt.subplot(2,2,1)
+    plt.imshow(arr)
+    plt.subplot(2,2,2)
+    plt.imshow(arr2)
+    plt.show()
+
+    plt.subplot(2,2,1)
+    plt.imshow(arr3)
+    plt.colorbar()
+    plt.subplot(2,2,2)
+    plt.imshow(arr4)
+    plt.colorbar()
+    plt.show()
+    
+
+    return
+
+if(__name__=="__main__"):
+    test_1()
+
+    exit(0)
+