From 016e2f96d59e212affac9b9fcad12b13178be3b0 Mon Sep 17 00:00:00 2001
From: Tom Pollard <tpollard@mit.edu>
Date: Mon, 8 Jan 2024 17:40:01 +0000
Subject: [PATCH] from_pydict -> from_pylist.

---
 tests/test_schema.py | 44 +++++++++++++++++++++++++-------------------
 1 file changed, 25 insertions(+), 19 deletions(-)

diff --git a/tests/test_schema.py b/tests/test_schema.py
index 1e69b6c..c9ec5f2 100644
--- a/tests/test_schema.py
+++ b/tests/test_schema.py
@@ -11,34 +11,40 @@ def test_patient_schema():
     """
     Test that mock patient data follows the patient_schema schema.
     """
-    patient_data = {
-        "patient_id": [123],
-        "events": [[{  # Nested list for events
-            "time": datetime.datetime(2020, 1, 1, 12, 0, 0),
-            "measurements": [{  # Nested list for measurements
-                "code": "some_code",
-                "text_value": "Example",
-                "numeric_value": 10.0,
-                "datetime_value": datetime.datetime(2020, 1, 1, 12, 0, 0),
-                "metadata": None
+    # Each element in the list is a row in the table
+    patient_data = [
+        {
+            "patient_id": 123,
+            "events": [{  # Nested list for events
+                "time": datetime.datetime(2020, 1, 1, 12, 0, 0),
+                "measurements": [{  # Nested list for measurements
+                    "code": "some_code",
+                    "text_value": "Example",
+                    "numeric_value": 10.0,
+                    "datetime_value": datetime.datetime(2020, 1, 1, 12, 0, 0),
+                    "metadata": None
+                }]
             }]
-        }]]
-    }
+        }
+    ]
 
-    patient_table = pa.Table.from_pydict(patient_data, schema=patient_schema())
+    patient_table = pa.Table.from_pylist(patient_data, schema=patient_schema())
     assert patient_table.schema.equals(patient_schema()), "Patient schema does not match"
 
 def test_label_schema():
     """
     Test that mock label data follows the label schema.
     """
-    label_data = {
-        "patient_id": [123],
-        "prediction_time": [datetime.datetime(2020, 1, 1, 12, 0, 0)],
-        "boolean_value": [True]
-    }
+    # Each element in the list is a row in the table
+    label_data = [
+        {
+            "patient_id": 123,
+            "prediction_time": datetime.datetime(2020, 1, 1, 12, 0, 0),
+            "boolean_value": True
+        }
+    ]
 
-    label_table = pa.Table.from_pydict(label_data, schema=label)
+    label_table = pa.Table.from_pylist(label_data, schema=label)
     assert label_table.schema.equals(label), "Label schema does not match"
 
 def test_dataset_metadata_schema():