def from_nwb(cls, nwbfile: NWBFile, **kwargs) -> "BehaviorSession":
behavior_session_id = BehaviorSessionId.from_nwb(nwbfile)
stimulus_timestamps = StimulusTimestamps.from_nwb(nwbfile)
running_acquisition = RunningAcquisition.from_nwb(nwbfile)
raw_running_speed = RunningSpeed.from_nwb(nwbfile, filtered=False)
running_speed = RunningSpeed.from_nwb(nwbfile)
metadata = BehaviorMetadata.from_nwb(nwbfile)
licks = Licks.from_nwb(nwbfile=nwbfile)
rewards = Rewards.from_nwb(nwbfile=nwbfile)
stimuli = Stimuli.from_nwb(nwbfile=nwbfile)
task_parameters = TaskParameters.from_nwb(nwbfile=nwbfile)
trials = TrialTable.from_nwb(nwbfile=nwbfile)
date_of_acquisition = DateOfAcquisition.from_nwb(nwbfile=nwbfile)
return BehaviorSession(behavior_session_id=behavior_session_id,
stimulus_timestamps=stimulus_timestamps,
running_acquisition=running_acquisition,
raw_running_speed=raw_running_speed,
running_speed=running_speed,
metadata=metadata,
licks=licks,
rewards=rewards,
stimuli=stimuli,
task_parameters=task_parameters,
trials=trials,
date_of_acquisition=date_of_acquisition)
def test_running_acquisition_nwb_roundtrip(nwbfile,
data_object_roundtrip_fixture,
roundtrip, running_acq_data):
running_acq = RunningAcquisition(running_acquisition=running_acq_data)
nwbfile = running_acq.to_nwb(nwbfile)
if roundtrip:
obt = data_object_roundtrip_fixture(nwbfile, RunningAcquisition)
else:
obt = RunningAcquisition.from_nwb(nwbfile)
pd.testing.assert_frame_equal(obt.value, running_acq_data, check_like=True)