adding MultiIndexValuesProvider
This commit is contained in:
13
src/MultiIndexValuesProvider.py
Normal file
13
src/MultiIndexValuesProvider.py
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
class MultiIndexValuesProvider:
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def getValues(multiIndex):
|
||||||
|
return set(MultiIndexValuesProvider._concat(MultiIndexValuesProvider._tuples2Lists(multiIndex.values)))
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _tuples2Lists(tuples):
|
||||||
|
return [list(tuple) for tuple in tuples]
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _concat(lsts):
|
||||||
|
return sum(lsts, [])
|
||||||
18
src/MultiIndexValuesProviderTest.py
Normal file
18
src/MultiIndexValuesProviderTest.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
import unittest
|
||||||
|
from MultiIndexValuesProvider import MultiIndexValuesProvider
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
class MultiIndexValuesProviderTest(unittest.TestCase):
|
||||||
|
|
||||||
|
def test_getValues(self):
|
||||||
|
# Given
|
||||||
|
multiIndex = pd.MultiIndex.from_tuples(
|
||||||
|
names = ['VAX_LOT1', 'VAX_LOT2'],
|
||||||
|
tuples = [['1808982', 'EW0175'],
|
||||||
|
['1808982', 'EW0176']])
|
||||||
|
|
||||||
|
# When
|
||||||
|
values = MultiIndexValuesProvider.getValues(multiIndex)
|
||||||
|
|
||||||
|
# Then
|
||||||
|
self.assertEqual(values, {'1808982', 'EW0175', 'EW0176'})
|
||||||
Reference in New Issue
Block a user