refactoring

This commit is contained in:
frankknoll
2023-02-10 08:30:41 +01:00
parent 9256c80e61
commit 8f1eb03d9b

View File

@@ -6,13 +6,16 @@ class MultiIndexValuesProviderTest(unittest.TestCase):
def test_getValues(self): def test_getValues(self):
# Given # Given
value1 = '1808982'
value2 = 'EW0175'
value3 = 'EW0176'
multiIndex = pd.MultiIndex.from_tuples( multiIndex = pd.MultiIndex.from_tuples(
names = ['VAX_LOT1', 'VAX_LOT2'], names = ['VAX_LOT1', 'VAX_LOT2'],
tuples = [['1808982', 'EW0175'], tuples = [[value1, value2],
['1808982', 'EW0176']]) [value1, value3]])
# When # When
values = MultiIndexValuesProvider.getValues(multiIndex) values = MultiIndexValuesProvider.getValues(multiIndex)
# Then # Then
self.assertEqual(values, {'1808982', 'EW0175', 'EW0176'}) self.assertEqual(values, {value1, value2, value3})