From 7db1613b1ae0da7ab1073d4cb6d5e91bf1053fb3 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Thu, 6 Feb 2025 08:25:59 -0800 Subject: [PATCH] test: Add provider-prefixed deepseek model test cases --- tests/basic/test_models.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index f39d62e0d..a05b6b559 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -212,6 +212,21 @@ class TestModels(unittest.TestCase): self.assertFalse(model.use_temperature) self.assertEqual(model.remove_reasoning, "think") + # Test provider/deepseek-r1 case + model = Model("someprovider/deepseek-r1") + self.assertEqual(model.edit_format, "diff") + self.assertTrue(model.use_repo_map) + self.assertTrue(model.examples_as_sys_msg) + self.assertFalse(model.use_temperature) + self.assertEqual(model.remove_reasoning, "think") + + # Test provider/deepseek-v3 case + model = Model("anotherprovider/deepseek-v3") + self.assertEqual(model.edit_format, "diff") + self.assertTrue(model.use_repo_map) + self.assertEqual(model.reminder, "sys") + self.assertTrue(model.examples_as_sys_msg) + # Test llama3 70b case model = Model("llama3-70b") self.assertEqual(model.edit_format, "diff")