sanity check for df_events and fip_df test

AllenNeuralDynamics · Nov 1, 2024 · ac84076 · ac84076
1 parent 5b447ae
commit ac84076
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/aind_dynamic_foraging_data_utils/nwb_utils.py b/src/aind_dynamic_foraging_data_utils/nwb_utils.py
@@ -514,6 +514,10 @@ def create_events_df(nwb_filename, adjust_time=True):
             trial_index.append(starts[-1])
     df["trial"] = trial_index
 
+    # Sanity check that the first go cue is time 0
+    gocues = df.query('event == "goCue_start_time"')
+    if (len(gocues) > 0) and (adjust_time):
+        assert np.isclose(gocues.iloc[0]['timestamps'], 0, rtol=0.01)
     return df