def test_empty_presentationwise_spike_times(spike_times_api):
# Test that when there are no spikes presentationwise_spike_times doesn't fail and instead returns a empty dataframe
spike_times_api.get_spike_times = types.MethodType(get_no_spikes_times, spike_times_api)
session = EcephysSession(api=spike_times_api)
obtained = session.presentationwise_spike_times(session.stimulus_presentations.index.values,
session.units.index.values)
assert(isinstance(obtained, pd.DataFrame))
assert(obtained.empty)
def test_presentationwise_spike_times(spike_times_api):
session = EcephysSession(api=spike_times_api)
obtained = session.presentationwise_spike_times(session.stimulus_presentations.index.values, session.units.index.values)
expected = pd.DataFrame({
'unit_id': [2, 2, 2],
'stimulus_presentation_id': [2, 2, 2, ],
'time_since_stimulus_presentation_onset': [0.01, 0.02, 0.03]
}, index=pd.Index(name='spike_time', data=[1.01, 1.02, 1.03]))
pd.testing.assert_frame_equal(expected, obtained, check_like=True, check_dtype=False)