Skip to content

Commit 52df793

Browse files
committed
Added simple integration test component -- test is currently broken
1 parent d3ef0ae commit 52df793

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/test_meds.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,14 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]:
304304
code: {regex: DISCHARGE.*}
305305
death:
306306
code: DEATH
307+
over_ninety_eight:
308+
code: null
309+
value_min: 98
310+
value_min_inclusive: False
311+
code_is_HR:
312+
code: LAB//HR
313+
high_HR:
314+
expr: and(code_is_HR, over_ninety_eight)
307315
discharge_or_death:
308316
expr: or(discharge, death)
309317
@@ -373,6 +381,7 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]:
373381
pl.Field("admission", pl.Int64),
374382
pl.Field("discharge", pl.Int64),
375383
pl.Field("death", pl.Int64),
384+
pl.Field("high_HR", pl.Int64),
376385
pl.Field("discharge_or_death", pl.Int64),
377386
pl.Field("_ANY_EVENT", pl.Int64),
378387
]
@@ -429,6 +438,7 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]:
429438
"admission": 0,
430439
"discharge": 0,
431440
"death": 0,
441+
"high_HR": 0,
432442
"discharge_or_death": 0,
433443
"_ANY_EVENT": 4
434444
},

0 commit comments

Comments
 (0)