Add tests for groq-llama2 and groq-mixtral models

haseeb-heaven · Feb 28, 2024 · 106e32e · 106e32e
1 parent 001498e
commit 106e32e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_interpreter.py b/tests/test_interpreter.py
@@ -47,6 +47,16 @@ def test_interpreter_gpt_4_model(self):
         args = Namespace(exec=True, save_code=True, mode='code', model='gpt-4', display_code=True, lang='python')
         interpreter = Interpreter(args)
         self.assertEqual(interpreter.args.model, 'gpt-4')
+
+    def test_interpreter_groq_llama2_model(self):
+        args = Namespace(exec=True, save_code=True, mode='code', model='groq-llama2', display_code=True, lang='python')
+        interpreter = Interpreter(args)
+        self.assertEqual(interpreter.args.model, 'groq-llama2')
+
+    def test_interpreter_groq_mixtral_model(self):
+        args = Namespace(exec=True, save_code=True, mode='code', model='groq-mixtral', display_code=True, lang='python')
+        interpreter = Interpreter(args)
+        self.assertEqual(interpreter.args.model, 'groq-mixtral')
 
     def test_interpreter_claude_2_model(self):
         args = Namespace(exec=True, save_code=True, mode='code', model='claude-2', display_code=True, lang='python')