Jelajahi Sumber

added test-data for float values

Oliver Kloß 2 tahun lalu
induk
melakukan
d70c79d66c

TEMPAT SAMPAH
unittest/spike_train_dissimilarity/victor_purpura_distance/data/matlab_results.mat


TEMPAT SAMPAH
unittest/spike_train_dissimilarity/victor_purpura_distance/data/matlab_results.npy


TEMPAT SAMPAH
unittest/spike_train_dissimilarity/victor_purpura_distance/data/times.mat


TEMPAT SAMPAH
unittest/spike_train_dissimilarity/victor_purpura_distance/data/times.npy


+ 14 - 12
unittest/spike_train_dissimilarity/victor_purpura_distance/scripts/generate_test_data_spike_train_dissimilarity.py

@@ -6,23 +6,25 @@ import scipy.io
 
 def generate_data():
 
-    st = []
-    times = []
+    times_int = []
+    times_float = []
 
     np.random.seed(1138)
     for i in range(3):
-        times.append([np.random.randint(0, 100) for x in range(1, 21)])
-        st.append(SpikeTrain(times[i], units='ms', t_start=0*pq.ms, t_stop=1000*pq.ms))
+        times_int.append(np.random.randint(0, 100, 20).tolist())
+        times_float.append([round(np.random.uniform(0, 100), 3) for x in range(20)])
 
-    print(times)
-    #print(st)
-    times_np = np.array(times)
-    #print(type(times_np))
+    for elem in times_int:
+        elem = elem.sort()
+    for elem in times_float:
+        elem = elem.sort()
 
-    return times_np
+    print(times_int)
+    print(times_float)
+    return times_int, times_float
 
-times = generate_data()
+times_int, times_float = generate_data()
 
 data_folder = '../data/'
-np.save(data_folder + 'times.npy', times)   #delete file before saving
-scipy.io.savemat(data_folder + 'times.mat', {"times": times})
+np.save(data_folder + 'times.npy', times_int, times_float)   #delete file before saving
+scipy.io.savemat(data_folder + 'times.mat', {"times_int": times_int, "times_float": times_float})

+ 10 - 7
unittest/spike_train_dissimilarity/victor_purpura_distance/scripts/generate_test_results.py

@@ -3,14 +3,17 @@ import scipy.io
 
 def generate_data():
 
-    results = [[0, 36, 36],
-               [36, 0, 34],
-               [36, 34, 0]]
+    results_int = [[0, 28, 29],
+                   [28, 0, 27],
+                   [29, 27, 0]]
+    results_float = [[0, 35.472, 28.242],
+                     [35.472, 0, 27.334],
+                     [28.242, 27.334, 0]]
 
-    return results
+    return results_int, results_float
 
-res = generate_data()
+results_int, results_float = generate_data()
 
 data_folder = '../data/'
-np.save(data_folder + 'matlab_results.npy', res)
-scipy.io.savemat(data_folder + 'matlab_results.mat', {"res": res})
+np.save(data_folder + 'matlab_results.npy', results_int, results_float)
+scipy.io.savemat(data_folder + 'matlab_results.mat', {"results_int": results_int, "results_float": results_float})