refactor: Refactor imports to be compatible with unit tests

main.py

@@ -0,0 +1 @@
+from scripts.main import main

scripts/browse.py

@@ -1,7 +1,7 @@
 import requests
 from bs4 import BeautifulSoup
-from config import Config
-from llm_utils import create_chat_completion
+from .config import Config
+from .llm_utils import create_chat_completion
 
 cfg = Config()
 

scripts/call_ai_function.py

@@ -1,8 +1,8 @@
-from config import Config
+from .config import Config
 
 cfg = Config()
 
-from llm_utils import create_chat_completion
+from .llm_utils import create_chat_completion
 # This is a magic function that can do anything with no-code. See
 # https://github.com/Torantulino/AI-Functions for more info.
 def call_ai_function(function, args, description, model=None):

scripts/json_parser.py

@@ -1,8 +1,8 @@
 import json
 from typing import Any, Dict, Union
-from call_ai_function import call_ai_function
-from config import Config
-from json_utils import correct_json
+from .call_ai_function import call_ai_function
+from .config import Config
+from .json_utils import correct_json
 
 cfg = Config()
 

scripts/json_utils.py

@@ -1,6 +1,6 @@
 import re
 import json
-from config import Config
+from .config import Config
 
 cfg = Config()
 

scripts/llm_utils.py

@@ -1,5 +1,5 @@
 import openai
-from config import Config
+from .config import Config
 cfg = Config()
 
 openai.api_key = cfg.openai_api_key

tests.py

@@ -0,0 +1,8 @@
+import unittest
+
+if __name__ == "__main__":
+    # Load all tests from the 'scripts/tests' package
+    suite = unittest.defaultTestLoader.discover('scripts/tests')
+
+    # Run the tests
+    unittest.TextTestRunner().run(suite)

tests/context.py

@@ -0,0 +1,4 @@
+import sys
+import os
+sys.path.insert(0, os.path.abspath(
+    os.path.join(os.path.dirname(__file__), '..')))

tests/test_browse_scrape_text.py

@@ -2,7 +2,6 @@
 # Generated by CodiumAI
 
 import requests
-import pytest
 
 from scripts.browse import scrape_text
 

tests/json_tests.py → tests/test_json_parser.py

@@ -1,9 +1,6 @@
 import unittest
-import os
-import sys
-# Probably a better way:
-sys.path.append(os.path.abspath('../scripts'))
-from json_parser import fix_and_parse_json
+
+from scripts.json_parser import fix_and_parse_json
 
 class TestParseJson(unittest.TestCase):
     def test_valid_json(self):