diff --git a/tests/test_other_meds.py b/tests/test_other_meds.py index ae80c7e..71e763a 100644 --- a/tests/test_other_meds.py +++ b/tests/test_other_meds.py @@ -271,6 +271,13 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]: trigger: discharge windows: + data_within_5yr_of_admit: + start: end - 1825d + end: admission_is_HF.start + start_inclusive: True + end_inclusive: False + has: + _ANY_EVENT: (1, None) admission_is_HF: start: end <- admission end: trigger @@ -319,7 +326,6 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]: """ "0": |-2 patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value - 1,01/17/2020 10:00,0,,, 1,01/20/2022 08:00,1,,, "1": |-2