Update stubs

This commit is contained in:
Trinh Anh Ngoc
2024-06-13 22:59:08 +07:00
parent 930f9155cf
commit da66ee0204

View File

@@ -1,5 +1,6 @@
import collections import collections
from concurrent.futures import Future from concurrent.futures import Future
from contextlib import contextmanager
from itertools import count from itertools import count
from typing import Any, Callable, Generator, Generic, Iterator, Mapping, Match, TypeVar from typing import Any, Callable, Generator, Generic, Iterator, Mapping, Match, TypeVar
from unittest.mock import Mock from unittest.mock import Mock
@@ -83,12 +84,15 @@ class BaseCase(case.TestCase, metaclass=MetaCase):
def startClassPatcher(cls, patcher): ... def startClassPatcher(cls, patcher): ...
def with_user(self, login: str) -> None: ... def with_user(self, login: str) -> None: ...
def debug_mode(self) -> Generator[None, None, None]: ... def debug_mode(self) -> Generator[None, None, None]: ...
@contextmanager
def assertRaises( def assertRaises(
self, exception, func: Any | None = ..., *args, **kwargs self, exception, func: Any | None = ..., *args, **kwargs
) -> Generator[Any, None, None] | None: ... ) -> Generator[Any, None, None] | None: ...
@contextmanager
def assertQueries( def assertQueries(
self, expected, flush: bool = ... self, expected, flush: bool = ...
) -> Generator[list, None, None]: ... ) -> Generator[list, None, None]: ...
@contextmanager
def assertQueryCount( def assertQueryCount(
self, default: int = ..., flush: bool = ..., **counters self, default: int = ..., flush: bool = ..., **counters
) -> Generator[None, None, None]: ... ) -> Generator[None, None, None]: ...