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)