From 8f1eb03d9b59c9599f913e1ce6bddb13f931ad8a Mon Sep 17 00:00:00 2001 From: frankknoll Date: Fri, 10 Feb 2023 08:30:41 +0100 Subject: [PATCH] refactoring --- src/MultiIndexValuesProviderTest.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/MultiIndexValuesProviderTest.py b/src/MultiIndexValuesProviderTest.py index 9d08e830b16..0da5c527dc8 100644 --- a/src/MultiIndexValuesProviderTest.py +++ b/src/MultiIndexValuesProviderTest.py @@ -6,13 +6,16 @@ class MultiIndexValuesProviderTest(unittest.TestCase): def test_getValues(self): # Given + value1 = '1808982' + value2 = 'EW0175' + value3 = 'EW0176' multiIndex = pd.MultiIndex.from_tuples( names = ['VAX_LOT1', 'VAX_LOT2'], - tuples = [['1808982', 'EW0175'], - ['1808982', 'EW0176']]) + tuples = [[value1, value2], + [value1, value3]]) # When values = MultiIndexValuesProvider.getValues(multiIndex) # Then - self.assertEqual(values, {'1808982', 'EW0175', 'EW0176'}) + self.assertEqual(values, {value1, value2, value3})