From dd00dec648c96af6b90867c1f09b44432a3b070a Mon Sep 17 00:00:00 2001
From: Matthew McDermott <mattmcdermott8@gmail.com>
Date: Fri, 23 Aug 2024 17:31:19 -0400
Subject: [PATCH] Added expected output to new test ase

---
 tests/test_other_meds.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tests/test_other_meds.py b/tests/test_other_meds.py
index d3a9ca7..0074274 100644
--- a/tests/test_other_meds.py
+++ b/tests/test_other_meds.py
@@ -185,7 +185,7 @@ def parse_labels_yaml(yaml_str: str) -> dict[str, pl.DataFrame]:
     1,1/17/20 10:00,ICD9CM//428.9,,
     1,1/17/20 10:00,DISCHARGE//HOME,,
     1,1/18/22 14:46,ADMISSION//MEDICAL,,
-    1,1/20/22 15:18,DISCHARGE//HOME_AMA,,
+    1,1/20/22 8:00,DISCHARGE//HOME_AMA,,
     1,1/20/22 8:00,ICD9CM//428.41,,
     1,1/20/22 8:00,ICD9CM//451.1,,
     1,1/24/22 8:11,ADMISSION//ED,,
@@ -305,15 +305,22 @@ 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,1/15/20 15:14,0,,,
+    1,1/19/22 14:46,0,,,
+    1,1/25/22 8:11,1,,,
 
   "1": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value
+    3,3/19/24 16:54,0,,,
+    3,3/30/24 11:00,0,,,
     """
     ),
     "HF_derived_readmission": parse_labels_yaml(
         """
   "0": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value
+    1,1/17/20 10:00,0,,,
+    1,1/20/22 8:00,1,,,
 
   "1": |-2
     patient_id,prediction_time,boolean_value,integer_value,float_value,categorical_value