starting a fresh tidy

tests/test_main.py

@@ -1,12 +1,11 @@
-import pytest
-from unittest.mock import patch, mock_open, MagicMock
-import yaml
 import logging
-import atexit
-import sys
+from unittest.mock import MagicMock, mock_open, patch
+
+import yaml
 
-from main import set_up_logging, load_config
 import config.exit_codes as ec
+from main import load_config, set_up_logging
+
 
 # Test for set_up_logging function
 def test_set_up_logging_success():