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(os.path.join(output_test_dir, "spikes.png"))
spikes = Spikes(time,
values,
xlabel="xlabel",
ylabel="ylabel",
extended_spikes=True)
spikes.plot(os.path.join(output_test_dir, "spikes_extended.png"))
class TestSpikes(TestCasePlot):
def setUp(self):
folder = os.path.dirname(os.path.realpath(__file__))
self.time = np.load(os.path.join(folder, "data/t_test.npy"))
self.values = np.load(os.path.join(folder, "data/U_test.npy"))
self.output_test_dir = ".tests/"
if os.path.isdir(self.output_test_dir):
shutil.rmtree(self.output_test_dir)
os.makedirs(self.output_test_dir)
self.figureformat = ".png"
def tearDown(self):
if os.path.isdir(self.output_test_dir):
shutil.rmtree(self.output_test_dir)
def test_init_no_input(self):
self.spikes = Spikes()
self.assertIsInstance(self.spikes, Spikes)
def test_init_input(self):
self.spikes = Spikes(self.time, self.values)
self.assertEqual(self.spikes.nr_spikes, 12)
def test_find_spikes_default(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time, self.values)
self.assertEqual(self.spikes.nr_spikes, 12)
def test_find_spikes_auto(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time, self.values, threshold="auto")
self.assertEqual(self.spikes.nr_spikes, 12)
def test_find_spikes_extended(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time, self.values, extended_spikes=True)
self.assertEqual(self.spikes.nr_spikes, 12)
def test_find_spikes_default_auto_extended(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time,
self.values,
threshold="auto",
extended_spikes=True)
self.assertEqual(self.spikes.nr_spikes, 12)
def test_iter(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time, self.values)
for spike in self.spikes:
self.assertIsInstance(spike, Spike)
def test_len(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time, self.values)
self.assertEqual(len(self.spikes), 12)
def test_getitem(self):
self.spikes = Spikes()
self.spikes.find_spikes(self.time, self.values)
result = self.spikes[0]
self.assertIsInstance(result, Spike)
def test_plot(self):
self.spikes = Spikes(self.time,
self.values,
xlabel="xlabel",
ylabel="ylabel")
self.spikes.plot(os.path.join(self.output_test_dir, "spikes.png"))
self.plot_exists("spikes")
def test_plot_extended(self):
self.spikes = Spikes(self.time,
self.values,
xlabel="xlabel",
ylabel="ylabel",
extended_spikes=True)
self.spikes.plot(
os.path.join(self.output_test_dir, "spikes_extended.png"))
self.plot_exists("spikes_extended")
def plot_exists(self, name):
plot_file = os.path.join(self.output_test_dir, name + ".png")
self.assertTrue(os.path.isfile(plot_file))
