francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-03-09 22:25:40 +01:00
parent 483e64ef43
commit 27920d536f
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/VaccineDistributionByZipcodeSimplifierTest.py
View File

@@ -8,11 +8,13 @@ class VaccineDistributionByZipcodeSimplifierTest(unittest.TestCase):
def test_sumDoses(self): def test_sumDoses(self):
# Given # Given
doses_shipped1 = 300.0
doses_shipped2 = 400.0
vaccineDistributionByZipcode = TestHelper.createDataFrame( vaccineDistributionByZipcode = TestHelper.createDataFrame(
columns = ['PROVIDER_NAME', 'ZIPCODE_SHP', 'LOT_NUMBER', 'DOSES_SHIPPED'], columns = ['PROVIDER_NAME', 'ZIPCODE_SHP', 'LOT_NUMBER', 'DOSES_SHIPPED'],
data = [ ['@PHARMACY.COM', '97206-2314', 'FK9893', 300.0], data = [ ['@PHARMACY.COM', '97206-2314', 'FK9893', 300.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', 300.0], ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', doses_shipped1],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', 400.0], ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', doses_shipped2],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]]) ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]])
# When # When
@@ -24,5 +26,5 @@ class VaccineDistributionByZipcodeSimplifierTest(unittest.TestCase):
TestHelper.createDataFrame( TestHelper.createDataFrame(
columns = ['PROVIDER_NAME', 'ZIPCODE_SHP', 'LOT_NUMBER', 'DOSES_SHIPPED'], columns = ['PROVIDER_NAME', 'ZIPCODE_SHP', 'LOT_NUMBER', 'DOSES_SHIPPED'],
data = [ ['@PHARMACY.COM', '97206-2314', 'FK9893', 300.0], data = [ ['@PHARMACY.COM', '97206-2314', 'FK9893', 300.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', 700.0], ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', doses_shipped1 + doses_shipped2],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]])) ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]]))