diff --git a/src/SymptomsByBatchcodesTableFactory.py b/src/SymptomByBatchcodeTableFactory.py similarity index 72% rename from src/SymptomsByBatchcodesTableFactory.py rename to src/SymptomByBatchcodeTableFactory.py index 82de5b0a6c2..e6a61e47555 100644 --- a/src/SymptomsByBatchcodesTableFactory.py +++ b/src/SymptomByBatchcodeTableFactory.py @@ -1,20 +1,20 @@ import pandas as pd import numpy as np -class SymptomsByBatchcodesTableFactory: +class SymptomByBatchcodeTableFactory: @staticmethod - def createSymptomsByBatchcodesTable(VAERSVAX, VAERSSYMPTOMS): - index_columns = SymptomsByBatchcodesTableFactory._getIndexColumns(VAERSVAX) + def createSymptomByBatchcodeTable(VAERSVAX, VAERSSYMPTOMS): + index_columns = SymptomByBatchcodeTableFactory._getIndexColumns(VAERSVAX) symptomColumn = 'SYMPTOM' return pd.merge( - SymptomsByBatchcodesTableFactory._get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns), - SymptomsByBatchcodesTableFactory._getSymptomsTable(VAERSSYMPTOMS, symptomColumn), + SymptomByBatchcodeTableFactory._get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns), + SymptomByBatchcodeTableFactory._getSymptomsTable(VAERSSYMPTOMS, symptomColumn), on = 'VAERS_ID').set_index(index_columns)[[symptomColumn]] @staticmethod def _getIndexColumns(VAERSVAX): - return [f"VAX_LOT{num}" for num in range(1, SymptomsByBatchcodesTableFactory._getMaxNumShots(VAERSVAX) + 1)] + return [f"VAX_LOT{num}" for num in range(1, SymptomByBatchcodeTableFactory._getMaxNumShots(VAERSVAX) + 1)] @staticmethod def _getMaxNumShots(VAERSVAX): @@ -23,7 +23,7 @@ class SymptomsByBatchcodesTableFactory: @staticmethod def _get_VAERSVAX_WITH_VAX_LOTS(VAERSVAX, index_columns): return pd.concat( - [VAERSVAX, SymptomsByBatchcodesTableFactory._getVaxLotsTable(VAERSVAX, index_columns)], + [VAERSVAX, SymptomByBatchcodeTableFactory._getVaxLotsTable(VAERSVAX, index_columns)], axis='columns').reset_index().drop_duplicates(subset = ['VAERS_ID'] + index_columns) @staticmethod diff --git a/src/SymptomsByBatchcodesTableFactoryTest.py b/src/SymptomByBatchcodeTableFactoryTest.py similarity index 90% rename from src/SymptomsByBatchcodesTableFactoryTest.py rename to src/SymptomByBatchcodeTableFactoryTest.py index 1cc6d4b1549..68c2ee2f421 100644 --- a/src/SymptomsByBatchcodesTableFactoryTest.py +++ b/src/SymptomByBatchcodeTableFactoryTest.py @@ -1,14 +1,14 @@ import unittest from pandas.testing import assert_frame_equal from TestHelper import TestHelper -from SymptomsByBatchcodesTableFactory import SymptomsByBatchcodesTableFactory +from SymptomByBatchcodeTableFactory import SymptomByBatchcodeTableFactory import pandas as pd import numpy as np # FIXME: brauchen Normalisierung analog zu DataFrameNormalizer.removeUnknownBatchCodes(dataFrame) und DataFrameNormalizer.convertVAX_LOTColumnToUpperCase(dataFrame) -class SymptomsByBatchcodesTableFactoryTest(unittest.TestCase): +class SymptomByBatchcodeTableFactoryTest(unittest.TestCase): - def test_createSymptomsByBatchcodesTable(self): + def test_createSymptomByBatchcodeTable(self): # Given VAERSVAX = TestHelper.createDataFrame( columns = ['VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES'], @@ -35,11 +35,11 @@ class SymptomsByBatchcodesTableFactoryTest(unittest.TestCase): 2547730])) # When - symptomsByBatchcodesTable = SymptomsByBatchcodesTableFactory.createSymptomsByBatchcodesTable(VAERSVAX, VAERSSYMPTOMS) + symptomByBatchcodeTable = SymptomByBatchcodeTableFactory.createSymptomByBatchcodeTable(VAERSVAX, VAERSSYMPTOMS) # Then assert_frame_equal( - symptomsByBatchcodesTable, + symptomByBatchcodeTable, TestHelper.createDataFrame( columns = ['SYMPTOM'], data = [ ['Blood pressure orthostatic abnormal'], @@ -59,7 +59,7 @@ class SymptomsByBatchcodesTableFactoryTest(unittest.TestCase): names = ['VAX_LOT1', 'VAX_LOT2'], tuples = [['1808982', 'EW0175']] * 13))) - def test_createSymptomsByBatchcodesTable_two_patients_same_symptoms(self): + def test_createSymptomByBatchcodeTable_two_patients_same_symptoms(self): # Given VAERSVAX = TestHelper.createDataFrame( columns = ['VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES'], @@ -84,11 +84,11 @@ class SymptomsByBatchcodesTableFactoryTest(unittest.TestCase): 2547731])) # When - symptomsByBatchcodesTable = SymptomsByBatchcodesTableFactory.createSymptomsByBatchcodesTable(VAERSVAX, VAERSSYMPTOMS) + symptomByBatchcodeTable = SymptomByBatchcodeTableFactory.createSymptomByBatchcodeTable(VAERSVAX, VAERSSYMPTOMS) # Then assert_frame_equal( - symptomsByBatchcodesTable, + symptomByBatchcodeTable, TestHelper.createDataFrame( columns = ['SYMPTOM'], data = [ ['Blood pressure orthostatic abnormal'], @@ -99,7 +99,7 @@ class SymptomsByBatchcodesTableFactoryTest(unittest.TestCase): 'EW0175'])), check_dtype = False) - def test_createSymptomsByBatchcodesTable_two_patients_distinct_symptoms(self): + def test_createSymptomByBatchcodeTable_two_patients_distinct_symptoms(self): # Given VAERSVAX = TestHelper.createDataFrame( columns = ['VAX_TYPE', 'VAX_MANU', 'VAX_LOT', 'VAX_DOSE_SERIES'], @@ -134,11 +134,11 @@ class SymptomsByBatchcodesTableFactoryTest(unittest.TestCase): 2547744])) # When - symptomsByBatchcodesTable = SymptomsByBatchcodesTableFactory.createSymptomsByBatchcodesTable(VAERSVAX, VAERSSYMPTOMS) + symptomByBatchcodeTable = SymptomByBatchcodeTableFactory.createSymptomByBatchcodeTable(VAERSVAX, VAERSSYMPTOMS) # Then assert_frame_equal( - symptomsByBatchcodesTable, + symptomByBatchcodeTable, TestHelper.createDataFrame( columns = ['SYMPTOM'], data = [ ['Blood pressure orthostatic abnormal'],