change to not add .txt or add if not given

Author: devsetgo

dsg_lib/common_functions/folder_functions.py

@@ -37,14 +37,13 @@
 License: MIT
 """
 
+# from loguru import logger
+import logging as logger
 import re
 from datetime import datetime
 from pathlib import Path
 from typing import List, Tuple
 
-# from loguru import logger
-import logging as logger
-
 # Define the directory where the files are located
 directory_to__files: str = "data"
 file_directory = f"{directory_to__files}/csv"

tests/test_common_functions/test_file_functions/test_save_text.py

@@ -29,7 +29,7 @@ def test_save_text(self, mock_save_text):
         mock_save_text(file_name=file_name, data=text, root_folder=self.data_dir)
 
         # Check that file was created and contains the correct text
-        file_path = os.path.join(self.data_dir, "text", f"{file_name}.txt")
+        file_path = os.path.join(self.data_dir, "text", file_name)
         self.assertTrue(os.path.exists(file_path))
         with open(file_path, "r") as file:
             self.assertEqual(file.read(), text)