Fix unit tests for the frontend language part (#872)

2024-08-01 12:39:12 -07:00
parent aba6f51f88
commit 6f221d4ca0
11 changed files with 80 additions and 83 deletions
--- a/test/lang/test_litellm_backend.py
+++ b/test/lang/test_litellm_backend.py
@@ -6,15 +6,12 @@ from sglang.test.test_programs import test_mt_bench, test_stream


 class TestAnthropicBackend(unittest.TestCase):
-    backend = None
    chat_backend = None

-    def setUp(self):
-        cls = type(self)
-
-        if cls.backend is None:
-            cls.backend = LiteLLM("gpt-3.5-turbo")
-            set_default_backend(cls.backend)
+    @classmethod
+    def setUpClass(cls):
+        cls.chat_backend = LiteLLM("gpt-3.5-turbo")
+        set_default_backend(cls.chat_backend)

    def test_mt_bench(self):
        test_mt_bench()