test: fix

uc-cdis · Jan 15, 2025 · dea672c · dea672c
1 parent 3347918
commit dea672c
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/test_agg_mds_tcia_adapter.py b/tests/test_agg_mds_tcia_adapter.py
@@ -13,7 +13,7 @@ def test_get_metadata_tcia():
             "StudyDate": "",
             "StudyDescription": "Collection One.",
             "PatientAge": "",
-            "PatientID": "",
+            "PatientID": "patient_id_1",
             "PatientName": "",
             "PatientSex": "",
             "EthnicGroup": "",
@@ -27,7 +27,7 @@ def test_get_metadata_tcia():
             "StudyDate": "",
             "StudyDescription": "Collection Two.",
             "PatientAge": "",
-            "PatientID": "",
+            "PatientID": "patient_id_2",
             "PatientName": "",
             "PatientSex": "",
             "EthnicGroup": "",
@@ -44,6 +44,10 @@ def test_get_metadata_tcia():
         "commons": "TCIA",
         "study_title": "path:StudyDescription",
         "program_name": "path:Collection",
+        "patient_id": "path:PatientID",
+        "ethnic_group": "path:EthnicGroup", 
+        "gender": "path:PatientSex",
+        "series_count": "path:SeriesCount",
         "description": "",
         "tags": [],
     }
@@ -144,6 +148,10 @@ def test_get_metadata_tcia():
                 "commons": "TCIA",
                 "description": "TCIA data from collection: Collection1.",
                 "program_name": "Collection1",
+                "patient_id": "patient_id_1",
+                "ethnic_group": "", 
+                "gender": "",
+                "series_count": 1,
                 "study_title": "Collection One.",
                 "tags": [{"category": "program_name", "name": "Collection1"}],
             },
@@ -155,6 +163,10 @@ def test_get_metadata_tcia():
                 "commons": "TCIA",
                 "description": "TCIA data from collection: Collection2.",
                 "program_name": "Collection2",
+                "patient_id": "patient_id_2",
+                "ethnic_group": "", 
+                "gender": "",
+                "series_count": 2,
                 "study_title": "Collection Two.",
                 "tags": [{"category": "program_name", "name": "Collection2"}],
             },