from elephant.spike_train_generation import             homogeneous_poisson_process, homogeneous_gamma_process
import quantities as pq
import matplotlib.pyplot as plt
from viziphant.rasterplot import rasterplot_rates

spiketrains1 = [homogeneous_poisson_process(rate=10 * pq.Hz)
                for _ in range(100)]
spiketrains2 = [homogeneous_gamma_process(a=3, b=10 * pq.Hz)
                for _ in range(100)]
rasterplot_rates([spiketrains1, spiketrains2])
plt.show()