adding MultiIndexExploderTest

2023-02-14 16:51:22 +01:00
parent 0ae6e56f33
commit 151aa9cd48
3 changed files with 63 additions and 0 deletions
--- a/src/MultiIndexExploderTest.py
+++ b/src/MultiIndexExploderTest.py
@@ -0,0 +1,47 @@
+import unittest
+from pandas.testing import assert_frame_equal
+from MultiIndexExploder import MultiIndexExploder
+from TestHelper import TestHelper
+import pandas as pd
+
+class MultiIndexExploderTest(unittest.TestCase):
+
+    def test_explodeMultiIndexOfTable(self):
+        # Given
+        table = TestHelper.createDataFrame(
+                columns = ['DATA'],
+                data = [  ['A, B data'],
+                          ['C, A data'],
+                          ['C, B data']],
+                index = pd.MultiIndex.from_tuples(
+                    names =   ['VAX_LOT1', 'VAX_LOT2'],
+                    tuples = [['A',        'B'],
+                              ['C',        'A'],
+                              ['C',        'B']]))
+
+        # When
+        explodedTable = MultiIndexExploder.explodeMultiIndexOfTable(table)
+
+        # Then
+        assert_frame_equal(
+            explodedTable,
+            TestHelper.createDataFrame(
+                columns = ['DATA'],
+                data = [  ['A, B data'],
+                          ['A, B data'],
+
+                          ['C, A data'],
+                          ['C, A data'],
+                          
+                          ['C, B data'],
+                          ['C, B data']],
+                index = pd.MultiIndex.from_tuples(
+                    names =   ['VAX_LOT_EXPLODED', 'VAX_LOT1', 'VAX_LOT2'],
+                    tuples = [['A',                'A',        'B'],
+                              ['B',                'A',        'B'],
+
+                              ['C',                'C',        'A'],
+                              ['A',                'C',        'A'],
+
+                              ['C',                'C',        'B'],                              
+                              ['B',                'C',        'B']])))