import numpy as np
import quantities as pq
import matplotlib.pyplot as plt
from elephant import spade
from elephant.spike_train_generation import homogeneous_poisson_process
import viziphant

np.random.seed(5)
spiketrains = [homogeneous_poisson_process(rate=1 * pq.Hz,
               t_stop=10 * pq.s) for _ in range(10)]
patterns = spade.spade(spiketrains, bin_size=400 * pq.ms,
                       winlen=1)['patterns']

axes = viziphant.patterns.plot_patterns(spiketrains, patterns[:2])
plt.show()