Debug and update unit tests.

AllenInstitute · Aug 14, 2023 · 3deb1e5 · 3deb1e5
1 parent def3732
commit 3deb1e5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/allensdk/brain_observatory/behavior/data_objects/stimuli/presentations.py b/allensdk/brain_observatory/behavior/data_objects/stimuli/presentations.py
@@ -261,8 +261,9 @@ def from_nwb(
                 stimulus_presentations=table
             )
         trials = None
-        if add_trials_dependent_values:
+        if add_trials_dependent_values and nwbfile.trials is not None:
             trials = Trials.from_nwb(nwbfile)
+
         return Presentations(presentations=table,
                              column_list=column_list,
                              trials=trials)
@@ -420,7 +421,9 @@ def from_stimulus_file(
             )
 
         return Presentations(
-            presentations=stim_pres_df, column_list=column_list
+            presentations=stim_pres_df,
+            column_list=column_list,
+            trials=trials
         )
 
     @classmethod

diff --git a/allensdk/test/brain_observatory/behavior/data_objects/test_stimuli.py b/allensdk/test/brain_observatory/behavior/data_objects/test_stimuli.py
@@ -52,6 +52,8 @@ def data(self):
                     {
                         "start_time": [300.0, 330.0, 360.0],
                         "stop_time": [330.0, 360.0, 360.0],
+                        "catch": [False, True, False],
+                        "change_frame": [-99, 99, -99] 
                     }
                 )