code: def test_add_shifted_events(daily_event_df_dict):
      """Tests `expand_holidays_with_lags` function."""
      shifted_events_dict = add_shifted_events(
          daily_event_df_dict=daily_event_df_dict,
          shifted_effect_lags=["1D", "-1D"],
      )
      new_daily_event_df_dict = shifted_events_dict[
          "new_daily_event_df_dict"
      ]
      shifted_events_cols = shifted_events_dict["shifted_events_cols"]
      drop_pred_cols = shifted_events_dict["drop_pred_cols"]
      assert sorted(new_daily_event_df_dict.keys()) == [
          "Christmas Day",
          "Christmas Day_1D_after",
          "Christmas Day_1D_before",
          "New Years Day",
          "New Years Day_1D_after",
          "New Years Day_1D_before",
      ]
      assert sorted(shifted_events_cols) == [
          "events_Christmas Day_1D_after",
          "events_Christmas Day_1D_before",
          "events_New Years Day_1D_after",
          "events_New Years Day_1D_before",
      ]
      assert sorted(drop_pred_cols) == [
          "C(Q('events_Christmas Day_1D_after'), levels=['', 'Christmas Day_1D_after'])",
          "C(Q('events_Christmas Day_1D_before'), levels=['', 'Christmas Day_1D_before'])",
          "C(Q('events_New Years Day_1D_after'), levels=['', 'New Years Day_1D_after'])",
          "C(Q('events_New Years Day_1D_before'), levels=['', 'New Years Day_1D_before'])",
      ]
  
file_path: greykite/greykite/tests/algo/common/test_holiday_utils.py
imports: import pandas
  import pytest
  from greykite.algo.common.holiday_utils import add_shifted_events
  from greykite.algo.common.holiday_utils import get_autoreg_holiday_interactions
  from greykite.algo.common.holiday_utils import get_dow_grouped_suffix
  from greykite.algo.common.holiday_utils import get_weekday_weekend_suffix
  from greykite.common.constants import EVENT_DF_DATE_COL
  from greykite.common.constants import EVENT_DF_LABEL_COL
label_desc: Use image node_id 5 for calling a global function with example usage: test_add_shifted_events(daily_event_df_dict) without return types