update tests

Author: neuromusic

tests/translator/foraging/test_foraging_translator.py

@@ -27,6 +27,8 @@ def test_load_time(behavioral_session_output_fixture):
 def test_load_trials(behavioral_session_output_fixture, trials_df_fixture):
     trials = foraging.load_trials(behavioral_session_output_fixture)
 
+    trials_df_fixture['change_frame'] += 1
+
     pd.testing.assert_frame_equal(
         trials,
         trials_df_fixture,

visual_behavior/translator/foraging/__init__.py

@@ -188,6 +188,7 @@ def stringify(x):
         trials[col] = trials[col].map(stringify)
 
     trials['change_frame'] = trials['change_frame'].map(lambda x: int(x) if np.isfinite(x) else None) + 1
+
     trials["change_image_category"] = trials["change_image_category"].apply(lambda x: x if x else '')  # use empty string instead of NoneType
     trials["change_image_name"] = trials["change_image_name"].apply(lambda x: x if x else '')  # use empty string instead of NoneType
     trials["initial_image_category"] = trials["initial_image_category"].apply(lambda x: x if x else '')  # use empty string instead of NoneType