def test_filter_owned_df_scalar(just_stimulus_table_api): session = EcephysSession(api=just_stimulus_table_api) ids = 3 obtained = session._filter_owned_df('stimulus_presentations', ids) assert obtained['phase'].values[0] == 180
def test_filter_owned_df(just_stimulus_table_api): session = EcephysSession(api=just_stimulus_table_api) ids = [0, 2] obtained = session._filter_owned_df('stimulus_presentations', ids) assert np.allclose([0, 120], obtained['phase'].values)