Example #1
0
 def test_get_spike_times_interval():
     ut = Units()
     ut.add_unit(spike_times=[0, 1, 2])
     ut.add_unit(spike_times=[3, 4, 5])
     np.testing.assert_array_equal(ut.get_unit_spike_times(0, (.5, 3)),
                                   [1, 2])
     np.testing.assert_array_equal(ut.get_unit_spike_times(0, (-.5, 1.1)),
                                   [0, 1])
Example #2
0
 def test_get_spike_times(self):
     ut = Units()
     ut.add_unit(spike_times=[0, 1, 2])
     ut.add_unit(spike_times=[3, 4, 5])
     self.assertTrue(all(ut.get_unit_spike_times(0) == np.array([0, 1, 2])))
     self.assertTrue(all(ut.get_unit_spike_times(1) == np.array([3, 4, 5])))
Example #3
0
 def test_get_spike_times_multi_interval(self):
     ut = Units()
     ut.add_unit(spike_times=[0, 1, 2])
     ut.add_unit(spike_times=[3, 4, 5])
     np.testing.assert_array_equal(ut.get_unit_spike_times((0, 1), (1.5, 3.5)), [[2], [3]])
Example #4
0
 def test_get_spike_times_multi(self):
     ut = Units()
     ut.add_unit(spike_times=[0, 1, 2])
     ut.add_unit(spike_times=[3, 4, 5])
     np.testing.assert_array_equal(ut.get_unit_spike_times((0, 1)), [[0, 1, 2], [3, 4, 5]])