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

adding VaccineDistributionByZipcodeSimplifier

Browse Source
This commit is contained in:
frankknoll
2023-03-09 20:38:17 +01:00
parent b089ca98f3
commit 483e64ef43
5 changed files with 62 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/VaccineDistributionByZipcodeSimplifierTest.py Normal file
View File

@@ -0,0 +1,28 @@
import unittest
from VaccineDistributionByZipcodeSimplifier import VaccineDistributionByZipcodeSimplifier
from TestHelper import TestHelper
from pandas.testing import assert_frame_equal
import pandas as pd
class VaccineDistributionByZipcodeSimplifierTest(unittest.TestCase):
def test_sumDoses(self):
# Given
vaccineDistributionByZipcode = TestHelper.createDataFrame(
columns = ['PROVIDER_NAME', 'ZIPCODE_SHP', 'LOT_NUMBER', 'DOSES_SHIPPED'],
data = [ ['@PHARMACY.COM', '97206-2314', 'FK9893', 300.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', 300.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', 400.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]])
# When
vaccineDistributionByZipcodeSimplified = VaccineDistributionByZipcodeSimplifier.sumDoses(vaccineDistributionByZipcode)
# Then
assert_frame_equal(
vaccineDistributionByZipcodeSimplified,
TestHelper.createDataFrame(
columns = ['PROVIDER_NAME', 'ZIPCODE_SHP', 'LOT_NUMBER', 'DOSES_SHIPPED'],
data = [ ['@PHARMACY.COM', '97206-2314', 'FK9893', 300.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', 700.0],
['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]]))