refactoring
This commit is contained in:
@@ -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})
|
||||||
|
|||||||
Reference in New Issue
Block a user