From 62e78eb1a38473174a6da162af00b30c127d9001 Mon Sep 17 00:00:00 2001 From: Piotr Mankowski Date: Sun, 9 Jun 2024 12:06:15 +0200 Subject: [PATCH] Small fix --- tests/test_ocl_loader.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/tests/test_ocl_loader.py b/tests/test_ocl_loader.py index 49b13a1..c57b9dd 100644 --- a/tests/test_ocl_loader.py +++ b/tests/test_ocl_loader.py @@ -1,23 +1,21 @@ - - import os import unittest from who_l3_smart_tools.core.terminology.ocl_loader import OclLoader -class TestOclLoader(unittest.testcase) +class TestOclLoader(unittest.testcase): def setUp(self): self.input_file = "../l3-data/test_dd.xlsx" self.output_dir = "../l3-data/output" - + def test_load_data(self): loader = OclLoader(self.input_file, self.output_dir) self.assertEqual(loader.load_data(), loader.df_dict) - + def test_transform_data(self): loader = OclLoader(self.input_file, self.output_dir) - + loader.transform_data() self.assertEqual(len(loader.df_dict["HIV.A"].columns), 9) @@ -27,4 +25,4 @@ def test_generate_ocl_csv(self): loader.transform_data() loader.generate_ocl_csv() - self.assertTrue(os.path.exists(f"{self.output_dir}/HIV.A.csv")) \ No newline at end of file + self.assertTrue(os.path.exists(f"{self.output_dir}/HIV.A.csv"))