1  # Python Imports
 2  import os
 3  import unittest
 4  from pathlib import Path
 5  
 6  from result import Ok
 7  
 8  # Project Imports
 9  from src.analysis.utils import file_utils
10  
11  
12  class TestFileUtils(unittest.TestCase):
13  
14      @classmethod
15      def setUpClass(cls):
16          cls.current_directory = os.path.dirname(os.path.abspath(__file__))
17  
18      def test_read_yaml_file(self):
19          test_file = os.path.join(self.current_directory, "resources/test.yaml")
20          result = file_utils.read_yaml_file(test_file)
21          expected_result = {"test": "test"}
22          self.assertEqual(result, expected_result)
23  
24      def test_get_files_from_folder_path(self):
25          test_folder = os.path.join(self.current_directory, "resources")
26          result = file_utils.get_files_from_folder_path(Path(test_folder))
27          expected_result = ["empty.yaml", "test.yaml"]
28          self.assertTrue(result.is_ok())
29          self.assertEqual(result.ok_value, expected_result)
30          self.assertIsInstance(result, Ok)
31  
32      def test_get_files_from_folder_path_error(self):
33          test_folder = os.path.join(self.current_directory, "resources", "not_exist")
34          result = file_utils.get_files_from_folder_path(Path(test_folder))
35          self.assertTrue(result.is_err())
36          self.assertEqual(f"{Path(test_folder)} does not exist.", result.err_value)