From 27015c2c624cecbc327219a44a9794bdae3e65f7 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sun, 25 Aug 2024 08:24:11 -0700 Subject: [PATCH] fix: Patch `requests.get` directly in tests --- tests/basic/test_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index 04102a78c..1ded9c1d5 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -3,6 +3,7 @@ import time import unittest from pathlib import Path from unittest.mock import mock_open, patch +import requests from aider.models import Model, get_model_info @@ -32,7 +33,7 @@ class TestModels(unittest.TestCase): @patch("aider.models.Path.stat") @patch("aider.models.safe_read_json") @patch("aider.models.safe_write_json") - @patch("aider.models.requests.get") + @patch("requests.get") def test_get_model_info(self, mock_get, mock_write_json, mock_read_json, mock_stat, mock_home): # Setup mock_home.return_value = Path("/mock/home")