refining SymptomByBatchcodeTableFactoryTest

2023-02-08 09:07:44 +01:00
parent f809f09a14
commit 408f523797
2 changed files with 55 additions and 54 deletions
--- a/src/SymptomByBatchcodeTableFactory.py
+++ b/src/SymptomByBatchcodeTableFactory.py
@@ -14,7 +14,7 @@ class SymptomByBatchcodeTableFactory:
                        SymptomByBatchcodeTableFactory._getSymptomsTable(VAERSSYMPTOMS, symptomColumn),
                        on = 'VAERS_ID')
                    .set_index(index_columns)
-                    [[symptomColumn]])
+                    [[symptomColumn, 'COUNTRY']])
    
    @staticmethod
    def _getIndexColumns(VAERSVAX):