From 5d2059d9d635d5118ac5e10e1d2e9e3dfcf68554 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Thu, 26 Jan 2023 08:59:46 +0100 Subject: [PATCH] making BatchCodeTableFactoryTest pass again --- src/BatchCodeTableFactoryTest.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/BatchCodeTableFactoryTest.py b/src/BatchCodeTableFactoryTest.py index f0630eab893..5762b48205c 100644 --- a/src/BatchCodeTableFactoryTest.py +++ b/src/BatchCodeTableFactoryTest.py @@ -28,7 +28,7 @@ class BatchCodeTableFactoryTest(unittest.TestCase): # Then assert_frame_equal( - batchCodeTable, + batchCodeTable[['Adverse Reaction Reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Company', 'Countries', 'Severe reports', 'Lethality']], TestHelper.createDataFrame( columns = ['Adverse Reaction Reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Company', 'Countries', 'Severe reports', 'Lethality'], data = [ [2, 1, 2, 2, 'MODERNA', 'France', 2/2 * 100, 1/2 * 100], @@ -62,7 +62,7 @@ class BatchCodeTableFactoryTest(unittest.TestCase): # Then assert_frame_equal( - batchCodeTable, + batchCodeTable[['Adverse Reaction Reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Company', 'Countries', 'Severe reports', 'Lethality']], TestHelper.createDataFrame( columns = ['Adverse Reaction Reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Company', 'Countries', 'Severe reports', 'Lethality'], data = [ [1, 1, 0, 0, 'PFIZER\BIONTECH', 'United Kingdom', 1/1 * 100, 1/1 * 100], @@ -98,7 +98,7 @@ class BatchCodeTableFactoryTest(unittest.TestCase): # Then assert_frame_equal( - batchCodeTable, + batchCodeTable[['Adverse Reaction Reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Company', 'Countries', 'Severe reports', 'Lethality']], TestHelper.createDataFrame( columns = ['Adverse Reaction Reports', 'Deaths', 'Disabilities', 'Life Threatening Illnesses', 'Company', 'Countries', 'Severe reports', 'Lethality'], data = [ ],