adding MultiIndexValuesProvider

This commit is contained in:
frankknoll
2023-02-09 15:49:15 +01:00
parent 16e6c2a729
commit 9256c80e61
2 changed files with 31 additions and 0 deletions

View 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, [])

View 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'})