From ed70a32671f7d1bc8875f46897747199627d91c8 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sat, 27 May 2023 04:39:47 -0700 Subject: [PATCH] Added import for MagicMock in test_main.py. --- tests/test_main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_main.py b/tests/test_main.py index 580e69d97..f97ba45c2 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,7 +1,8 @@ import os import tempfile import unittest -from unittest import TestCase, MagicMock +from unittest import TestCase +from unittest.mock import MagicMock from aider.main import main import subprocess from prompt_toolkit.input import create_input