Add test for flat_fields

globaldothealth · Aug 2, 2024 · 721fd72 · 721fd72
1 parent b951c78
commit 721fd72
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/test_base.py b/tests/test_base.py
@@ -1,9 +1,29 @@
 from fhirflat.resources.encounter import Encounter
+import fhirflat
 import pandas as pd
 import pytest
 from pydantic.v1 import ValidationError
 
 
+def test_flat_fields():
+    p = fhirflat.Patient()
+    ff = p.flat_fields()
+
+    assert ff == [
+        "id",
+        "extension",
+        "gender",
+        "birthDate",
+        "deceasedBoolean",
+        "deceasedDateTime",
+        "maritalStatus",
+        "multipleBirthBoolean",
+        "multipleBirthInteger",
+        "generalPractitioner",
+        "managingOrganization",
+    ]
+
+
 def test_validate_fhirflat_single_resource_errors():
     df = pd.DataFrame(
         {