VimPlug/typeshed
1
0
Fork 0
mirror of https://github.com/davidhalter/typeshed.git synced 2025-12-09 21:46:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix @patch when new is missing (#10459)

Browse Source
This commit is contained in:
Sebastian Rittau
2023-07-14 12:53:13 +02:00
committed by GitHub
parent 1d7f0d087b
commit 7ea173c4ad
2 changed files with 33 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
test_cases/stdlib/check_unittest.py
View File

@@ -5,7 +5,7 @@ from datetime import datetime, timedelta
from decimal import Decimal
from fractions import Fraction
from typing_extensions import assert_type
from unittest.mock import Mock, patch
from unittest.mock import MagicMock, Mock, patch
case = unittest.TestCase()
@@ -94,13 +94,20 @@ case.assertGreater(Bacon(), Ham()) # type: ignore
###
@patch("sys.exit", new=Mock())
def f(i: int) -> str:
@patch("sys.exit")
def f_default_new(i: int, mock: MagicMock) -> str:
return "asdf"
assert_type(f(1), str)
f("a") # type: ignore
@patch("sys.exit", new=42)
def f_explicit_new(i: int) -> str:
return "asdf"
assert_type(f_default_new(1), str)
f_default_new("a") # Not an error due to ParamSpec limitations
assert_type(f_explicit_new(1), str)
f_explicit_new("a") # type: ignore[arg-type]
@patch("sys.exit", new=Mock())