VimPlug/typeshed
1
0
Fork 0
mirror of https://github.com/davidhalter/typeshed.git synced 2025-12-10 05:51:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

TemporaryFile(): Handle one more case correctly (#10284)

Browse Source
This commit is contained in:
Sebastian Rittau
2023-06-14 14:33:01 +02:00
committed by GitHub
parent 6e90fce4b8
commit 5f9d05c7f5
2 changed files with 52 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
test_cases/stdlib/check_tempfile.py Normal file
View File

@@ -0,0 +1,31 @@
from __future__ import annotations
import io
import sys
from tempfile import TemporaryFile, _TemporaryFileWrapper
from typing_extensions import assert_type
if sys.platform == "win32":
assert_type(TemporaryFile(), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile("w+"), _TemporaryFileWrapper[str])
assert_type(TemporaryFile("w+b"), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile("wb"), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile("rb"), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile("wb", 0), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile(mode="w+"), _TemporaryFileWrapper[str])
assert_type(TemporaryFile(mode="w+b"), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile(mode="wb"), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile(mode="rb"), _TemporaryFileWrapper[bytes])
assert_type(TemporaryFile(buffering=0), _TemporaryFileWrapper[bytes])
else:
assert_type(TemporaryFile(), io.BufferedRandom)
assert_type(TemporaryFile("w+"), io.TextIOWrapper)
assert_type(TemporaryFile("w+b"), io.BufferedRandom)
assert_type(TemporaryFile("wb"), io.BufferedWriter)
assert_type(TemporaryFile("rb"), io.BufferedReader)
assert_type(TemporaryFile("wb", 0), io.FileIO)
assert_type(TemporaryFile(mode="w+"), io.TextIOWrapper)
assert_type(TemporaryFile(mode="w+b"), io.BufferedRandom)
assert_type(TemporaryFile(mode="wb"), io.BufferedWriter)
assert_type(TemporaryFile(mode="rb"), io.BufferedReader)
assert_type(TemporaryFile(buffering=0), io.FileIO)