Align tests to work after psf#966

florczakraf · Oct 2, 2019 · b2be34c · b2be34c
1 parent 36881c6
commit b2be34c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_black.py b/tests/test_black.py
@@ -1434,6 +1434,7 @@ def test_invalid_include_exclude(self) -> None:
         for option in ["--include", "--exclude"]:
             self.invokeBlack(["-", option, "**()(!!*)"], exit_code=2)
 
+    @event_loop(close=False)
     def test_check_shebang(self) -> None:
         test_path = THIS_DIR / "data" / "shebang_tests"
         empty_config = ["--config", str(THIS_DIR / "empty.toml")]
@@ -1448,6 +1449,7 @@ def test_check_shebang(self) -> None:
         self.assertNotIn(str(test_path / "decoding_error"), output)
         self.assertIn("\n1 file would be reformatted", output)
 
+    @event_loop(close=False)
     def test_custom_shebang_regex(self) -> None:
         test_path = THIS_DIR / "data" / "shebang_tests"
         empty_config = ["--config", str(THIS_DIR / "empty.toml")]