From a045bda1719b6efabd19f2a11c3578ce48edbfeb Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Wed, 30 Oct 2024 13:21:43 -0700
Subject: [PATCH] refactor: update test to use ModelInfoManager instead of
 get_model_info

---
 tests/basic/test_models.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index 24aad3d79..038e8024a 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -1,12 +1,13 @@
 import unittest
 from unittest.mock import ANY, MagicMock, patch
 
-from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models
+from aider.models import Model, ModelInfoManager, sanity_check_model, sanity_check_models
 
 
 class TestModels(unittest.TestCase):
     def test_get_model_info_nonexistent(self):
-        info = get_model_info("non-existent-model")
+        manager = ModelInfoManager()
+        info = manager.get_model_info("non-existent-model")
         self.assertEqual(info, {})
 
     def test_max_context_tokens(self):