elephant-data/unittest/spike_train_dissimilarity/victor_purpura_distance/scripts/generate_test_data_spike_train_dissimilarity.py

import numpy as np
import quantities as pq
from neo.core import SpikeTrain
import scipy.io


def generate_data():

    st = []
    times = []

    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))

    print(times)
    #print(st)
    times_np = np.array(times)
    #print(type(times_np))

    return times_np

times = 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})