示例#1
0
    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"))