def test_lists(self): train_size = 100 n = 10 senders = list(np.random.randint(n, size=train_size * n)) times = list(np.random.rand(train_size * n)) sp = PopulationSpikes(n, senders, times) # try to retrieve spike trains for nIdx in range(n): train = sp[nIdx] assert train is not None # Try to run all the methods, None should raise an exception sp.avg_firing_rate(0, 1) sp.sliding_firing_rate(0, 1, 0.05, 0.1) sp.windowed((0, 1)) sp.raster_data() sp.spike_train_difference(list(range(n)))
def test_empty(self): for N in range(0, 4): pop = PopulationSpikes(N, [], []) r, rt = pop.sliding_firing_rate(0, 1, .1, .2) assert np.all(r == 0)