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