patch: Test tgpt-auto

Simatwa · Apr 23, 2024 · a7fadcf · a7fadcf
1 parent cc512fc
commit a7fadcf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -39,4 +39,4 @@ jobs:
     - name: Test with Unittest
       run: |
         python -m unittest discover -s tests -p 'test_utils.py' -f -v
-#        python -m unittest discover -s tests -p 'test_gpt4free.py' -f -v */
+        python -m unittest discover -s tests -p 'test_auto.py' -f -v
diff --git a/tests/test_auto.py b/tests/test_auto.py
@@ -0,0 +1,12 @@
+import unittest
+import tests.base as base
+from pytgpt.auto import AUTO
+
+
+class TestAuto(base.llmBase):
+    def setUp(self):
+        self.bot = AUTO()
+        self.prompt = base.prompt
+
+if __name__ == "__main__":
+    unittest.main()