diff --git a/tests/test_get_wiki_images.py b/tests/test_get_wiki_images.py index 6dab678..4a7961f 100644 --- a/tests/test_get_wiki_images.py +++ b/tests/test_get_wiki_images.py @@ -222,13 +222,12 @@ def wikidata_response(self, image_data, vernacular_data): return {"url": url, "querystring": querystring, "response": response} def mock_patch_all_web_request_methods(self, f): - @mock.patch("requests.get", side_effect=self.mocked_requests_get) - @mock.patch( - "azure.ai.vision.imageanalysis.ImageAnalysisClient.analyze_from_url", - side_effect=self.mocked_analyze_from_url, - ) def functor(*args, **kwargs): - return f(*args, **kwargs) + with mock.patch("requests.get", side_effect=self.mocked_requests_get), mock.patch( + "azure.ai.vision.imageanalysis.ImageAnalysisClient.analyze_from_url", + side_effect=self.mocked_analyze_from_url, + ): + return f(*args, **kwargs) return functor