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

refining BarChartDescriptionTablesTest

Browse Source
This commit is contained in:
frankknoll
2023-08-27 12:36:54 +02:00
parent c77e1fe7c6
commit 64646bb9dc
2 changed files with 78 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/BarChartDescriptionTables.py
View File

@@ -21,6 +21,11 @@ class BarChartDescriptionTables:
sizeOfGuessedHistogram = sum(barChartDescription['Adverse Reaction Reports guessed'])
return sizeOfGuessedHistogram >= minSizeOfGuessedHistogram
@staticmethod
def hasMinSizeOfKnownHistogram(barChartDescription, minSizeOfKnownHistogram):
sizeOfKnownHistogram = sum(barChartDescription['Adverse Reaction Reports known'])
return sizeOfKnownHistogram >= minSizeOfKnownHistogram
@staticmethod
def hasCountryWithGuessedGreaterThanKnown(barChartDescription):
guessedBarChart = barChartDescription['Adverse Reaction Reports guessed']

73
src/BarChartDescriptionTablesTest.py
View File

@@ -137,6 +137,79 @@ class BarChartDescriptionTablesTest(unittest.TestCase):
name='VAX_LOT')),
check_dtype=True)
def test_filterHasMinSizeOfKnownHistogram_true(self):
# Given
barChartDescriptionTable = TestHelper.createDataFrame(
columns=['BAR_CHART_DESCRIPTION'],
data=[
[
{
'countries': ['Germany', 'Hungary'],
'Adverse Reaction Reports guessed': [10, 20],
'Adverse Reaction Reports known': [20, 30],
'Jensen-Shannon distance': 0.4711
}
]
],
index=pd.Index(
[
'!D0181'
],
name='VAX_LOT'))
# When
barChartDescriptionTableResult = BarChartDescriptionTables.filter(
barChartDescriptionTable,
lambda barChartDescription:
BarChartDescriptionTables.hasMinSizeOfKnownHistogram(
barChartDescription,
minSizeOfKnownHistogram=20))
# Then
assert_frame_equal(
barChartDescriptionTableResult,
barChartDescriptionTable,
check_dtype=True)
def test_filterHasMinSizeOfKnownHistogram_false(self):
# Given
barChartDescriptionTable = TestHelper.createDataFrame(
columns=['BAR_CHART_DESCRIPTION'],
data=[
[
{
'countries': ['Germany', 'Hungary'],
'Adverse Reaction Reports guessed': [100, 20],
'Adverse Reaction Reports known': [20, 30],
'Jensen-Shannon distance': 0.4711
}
]
],
index=pd.Index(
[
'!D0181'
],
name='VAX_LOT'))
# When
barChartDescriptionTableResult = BarChartDescriptionTables.filter(
barChartDescriptionTable,
lambda barChartDescription:
BarChartDescriptionTables.hasMinSizeOfKnownHistogram(
barChartDescription,
minSizeOfKnownHistogram=51))
# Then
assert_frame_equal(
barChartDescriptionTableResult,
TestHelper.createDataFrame(
columns=['BAR_CHART_DESCRIPTION'],
data=[],
index=pd.Index(
[],
name='VAX_LOT')),
check_dtype=True)
def test_filterHasCountryWithGuessedGreaterThanKnown(self):
# Given
guessed = 25