Consistently wrap two context managers in parens (in --preview). (#3589)

[etc/vim.git] / tests / data / preview_context_managers / targeting_py38.py

diff --git a/tests/data/preview_context_managers/targeting_py38.py b/tests/data/preview_context_managers/targeting_py38.py
index 6ec4684e441b8d891bd723483fcad2107d04ee5b..f125cdffb8af26e4d3cc78aa3b456809e7b6fb90 100644 (file)
--- a/tests/data/preview_context_managers/targeting_py38.py
+++ b/tests/data/preview_context_managers/targeting_py38.py
@@ -23,6 +23,14 @@ with \
     pass
 
 
+with mock.patch.object(
+    self.my_runner, "first_method", autospec=True
+) as mock_run_adb, mock.patch.object(
+    self.my_runner, "second_method", autospec=True, return_value="foo"
+):
+    pass
+
+
 # output
 
 
@@ -36,3 +44,11 @@ with make_context_manager1() as cm1, make_context_manager2(), make_context_manag
 
 with new_new_new1() as cm1, new_new_new2():
     pass
+
+
+with mock.patch.object(
+    self.my_runner, "first_method", autospec=True
+) as mock_run_adb, mock.patch.object(
+    self.my_runner, "second_method", autospec=True, return_value="foo"
+):
+    pass