Skip to content

Commit

Permalink
Small fix
Browse files Browse the repository at this point in the history
  • Loading branch information
pmanko committed Jun 9, 2024
1 parent 52815df commit 62e78eb
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions tests/test_ocl_loader.py
Original file line number Diff line number Diff line change
@@ -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)
Expand All @@ -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"))
self.assertTrue(os.path.exists(f"{self.output_dir}/HIV.A.csv"))

0 comments on commit 62e78eb

Please sign in to comment.