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)