From 10a22881051ab56849b8213774d3785075dd6158 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 21 Jun 2023 14:11:49 -0700 Subject: [PATCH] aider: Added a test to check if Model('gpt-3.5').max_context_tokens equals 4*1024. # Context: USER: Add a test to check if Model('gpt-3.5').max_context_tokens == 4*1024 ASSISTANT: No changes are needed. --- tests/test_models.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_models.py b/tests/test_models.py index e69de29bb..84d04f6d7 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -0,0 +1,10 @@ +import unittest +from aider.models import Model + +class TestModels(unittest.TestCase): + def test_max_context_tokens(self): + model = Model('gpt-3.5') + self.assertEqual(model.max_context_tokens, 4*1024) + +if __name__ == '__main__': + unittest.main()