def test_noisy(self): folder = os.path.dirname(os.path.realpath(__file__)) time = np.load(os.path.join(folder, "data/t_noise.npy")) values = np.load(os.path.join(folder, "data/V_noise.npy")) spikes = Spikes() spikes.find_spikes(time, values) spikes.plot_voltage("voltage.png") self.assertEqual(len(spikes), 16)
def generate_spikes_plot(): values = np.load(os.path.join(test_data_dir, "U_test.npy")) time = np.load(os.path.join(test_data_dir, "t_test.npy")) spikes = Spikes(time, values, xlabel="xlabel", ylabel="ylabel") spikes.plot_spikes(os.path.join(output_test_dir, "spikes.png")) spikes = Spikes(time, values, xlabel="xlabel", ylabel="ylabel") spikes.plot_voltage(os.path.join(output_test_dir, "U_test.png")) spikes = Spikes(time, values, xlabel="xlabel", ylabel="ylabel", extended_spikes=True) spikes.plot_spikes(os.path.join(output_test_dir, "spikes_extended.png")) values = np.load(os.path.join(test_data_dir, "V_spike.npy")) time = np.load(os.path.join(test_data_dir, "t_spike.npy")) spikes = Spikes(time, values, xlabel="xlabel", ylabel="ylabel") spikes.plot_voltage(os.path.join(output_test_dir, "V_spike.png")) values = np.load(os.path.join(test_data_dir, "V_noise.npy")) time = np.load(os.path.join(test_data_dir, "t_noise.npy")) spikes = Spikes(time, values, xlabel="xlabel", ylabel="ylabel") spikes.plot_voltage(os.path.join(output_test_dir, "V_noise.png"))