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

starting BarChartDescriptionTablesTest

Browse Source
This commit is contained in:
frankknoll
2023-08-26 20:12:08 +02:00
parent 5b179f0f1c
commit dc4fcd055c
4 changed files with 114 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
src/BarChartDescriptionTablesTest.py Normal file
View File

@@ -0,0 +1,62 @@
import unittest
import pandas as pd
import numpy as np
from pandas.testing import assert_frame_equal
from TestHelper import TestHelper
from BarChartDescriptionTables import BarChartDescriptionTables
class BarChartDescriptionTablesTest(unittest.TestCase):
def test_filterValidJensenShannonDistances(self):
# Given
barChartDescriptionTable = TestHelper.createDataFrame(
columns = ['BAR_CHART_DESCRIPTION'],
data = [
[
{
'countries': ['Germany', 'Hungary'],
'Adverse Reaction Reports guessed': [0, 0],
'Adverse Reaction Reports known': [20, 30],
'Jensen-Shannon distance': np.nan
}
],
[
{
'countries': ['Germany'],
'Adverse Reaction Reports guessed': [70],
'Adverse Reaction Reports known': [80],
'Jensen-Shannon distance': 0.4711
}
]
],
index = pd.Index(
[
'!D0181',
'# 009C01A'
],
name = 'VAX_LOT'))
# When
barChartDescriptionTableResult = BarChartDescriptionTables.filterValidJensenShannonDistances(barChartDescriptionTable)
# Then
assert_frame_equal(
barChartDescriptionTableResult,
TestHelper.createDataFrame(
columns = ['BAR_CHART_DESCRIPTION'],
data = [
[
{
'countries': ['Germany'],
'Adverse Reaction Reports guessed': [70],
'Adverse Reaction Reports known': [80],
'Jensen-Shannon distance': 0.4711
}
]
],
index = pd.Index(
[
'# 009C01A',
],
name = 'VAX_LOT')),
check_dtype = True)