francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

generating histograms for countries more efficient

Browse Source
This commit is contained in:
frankknoll
2023-02-16 14:59:43 +01:00
parent ef7c36d567
commit 0ffc5881b6
8 changed files with 139 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/MultiIndexExploderTest.py
View File

@@ -9,10 +9,10 @@ class MultiIndexExploderTest(unittest.TestCase):
def test_explodeMultiIndexOfTable(self):
# Given
table = TestHelper.createDataFrame(
columns = ['DATA'],
data = [ ['A, B data'],
['C, A data'],
['C, B data']],
columns = ['DATA', 'COUNTRY'],
data = [ ['A, B data', 'Country A'],
['C, A data', 'Country B'],
['C, B data', 'Country C']],
index = pd.MultiIndex.from_tuples(
names = ['VAX_LOT1', 'VAX_LOT2'],
tuples = [['A', 'B'],
@@ -26,15 +26,15 @@ class MultiIndexExploderTest(unittest.TestCase):
assert_frame_equal(
explodedTable,
TestHelper.createDataFrame(
columns = ['DATA'],
data = [ ['A, B data'],
['A, B data'],
columns = ['DATA', 'COUNTRY'],
data = [ ['A, B data', 'Country A'],
['A, B data', 'Country A'],
['C, A data'],
['C, A data'],
['C, A data', 'Country B'],
['C, A data', 'Country B'],
['C, B data'],
['C, B data']],
['C, B data', 'Country C'],
['C, B data', 'Country C']],
index = pd.MultiIndex.from_tuples(
names = ['VAX_LOT_EXPLODED', 'VAX_LOT1', 'VAX_LOT2'],
tuples = [['A', 'A', 'B'],