add spike-distance, PWL function (probably buggy)

1 files changed, 10 insertions, 2 deletions

diff --git a/examples/test_data.py b/examples/test_data.py
index bddcb15..94e7d51 100644
--- a/examples/test_data.py
+++ b/examples/test_data.py
@@ -16,13 +16,21 @@ for line in spike_file:
 for (i,spikes) in enumerate(spike_trains):
     plt.plot(spikes, i*np.ones_like(spikes), 'o')
 
-f = spk.isi_distance(spike_trains[0], spike_trains[1], 4000)
+f = spk.isi_distance(spike_trains[0], spike_trains[10], 4000)
 x, y = f.get_plottable_data()
 
 plt.figure()
-plt.plot(x, y, '-k')
+plt.plot(x, np.abs(y), '--k')
 
 print("Average: %.8f" % f.avrg())
 print("Absolute average: %.8f" % f.abs_avrg())
 
+
+f = spk.spike_distance(spike_trains[0], spike_trains[10], 4000)
+x, y = f.get_plottable_data()
+print(x)
+print(y)
+#plt.figure()
+plt.plot(x, y, '-b')
+
 plt.show()