From 27920d536f38dafa3d08b6d62b60d1ef5d078938 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Thu, 9 Mar 2023 22:25:40 +0100 Subject: [PATCH] refactoring --- src/VaccineDistributionByZipcodeSimplifierTest.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/VaccineDistributionByZipcodeSimplifierTest.py b/src/VaccineDistributionByZipcodeSimplifierTest.py index 63fe9e5889f..6273023f108 100644 --- a/src/VaccineDistributionByZipcodeSimplifierTest.py +++ b/src/VaccineDistributionByZipcodeSimplifierTest.py @@ -8,11 +8,13 @@ class VaccineDistributionByZipcodeSimplifierTest(unittest.TestCase): def test_sumDoses(self): # Given + doses_shipped1 = 300.0 + doses_shipped2 = 400.0 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', 'FK9893', doses_shipped1], + ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FK9893', doses_shipped2], ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]]) # When @@ -24,5 +26,5 @@ class VaccineDistributionByZipcodeSimplifierTest(unittest.TestCase): 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', 'FK9893', doses_shipped1 + doses_shipped2], ['@PHARMACY.COM - 82ND AVE', '97266-4885', 'FL8095', 200.0]]))