Skip to content

Commit 5980f6a

Browse files
committed
update tests
1 parent 451ebf9 commit 5980f6a

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tests/test_behavior.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,20 +48,20 @@ def test_constructors(self):
4848
session_end_time=now,
4949
animal_weight_prior=20.1,
5050
animal_weight_post=19.7,
51+
reward_consumed_total=1020,
5152
behavior_type="Foraging",
5253
stimulus_epochs=[
5354
bs.StimulusEpoch(
5455
stimulus=BehaviorStimulation(
5556
behavior_name="Foraging",
56-
behavior_software="Bonsai",
57-
session_number=3,
57+
behavior_software=["Bonsai"],
58+
session_number=[3],
5859
behavior_software_version="0.1",
5960
behavior_script="URL_to_code",
6061
behavior_script_version="0.1",
6162
input_parameters={"reward volume": 0.01},
6263
output_parameters={},
63-
reward_consumed_during_training=820,
64-
reward_consumed_total=1020,
64+
reward_consumed_during_epoch=820,
6565
trials_total=551,
6666
trials_finished=343,
6767
trials_rewarded=146,

tests/test_session.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ def test_constructors(self):
4949
manipulator_coordinates=Coordinates3d(x="1", y="1", z="1"),
5050
),
5151
],
52+
mouse_platform_name="Running wheel",
53+
active_mouse_platform=False,
5254
)
5355
],
5456
)

0 commit comments

Comments
 (0)