VimPlug/typeshed
1
0
Fork 0
mirror of https://github.com/davidhalter/typeshed.git synced 2025-12-07 04:34:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix the definition of nsmallest() in stdlib/2 and 2and3/heapq.pyi (#3114)

Browse Source 
Missing support of the optional "key" kwarg in nsmallest.
Also fixed nlargest syntax for 2and3 which was also missing.
Use a protocol for heapq.py
This commit is contained in:
Yannack
2019-07-18 15:38:28 -04:00
committed by Sebastian Rittau
parent 09d1055cd3
commit 40215d1fa3
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
stdlib/2/heapq.pyi
View File

@@ -1,7 +1,10 @@
from typing import TypeVar, List, Iterable, Any, Callable, Optional
from typing import TypeVar, List, Iterable, Any, Callable, Optional, Protocol
_T = TypeVar('_T')
class _Sortable(Protocol):
def __lt__(self: _T, other: _T) -> bool: ...
def cmp_lt(x, y) -> bool: ...
def heappush(heap: List[_T], item: _T) -> None: ...
def heappop(heap: List[_T]) -> _T:
@@ -12,5 +15,6 @@ def heapreplace(heap: List[_T], item: _T) -> _T:
raise IndexError() # if heap is empty
def merge(*iterables: Iterable[_T]) -> Iterable[_T]: ...
def nlargest(n: int, iterable: Iterable[_T],
key: Optional[Callable[[_T], Any]] = ...) -> List[_T]: ...
def nsmallest(n: int, iterable: Iterable[_T]) -> List[_T]: ...
key: Optional[Callable[[_T], _Sortable]] = ...) -> List[_T]: ...
def nsmallest(n: int, iterable: Iterable[_T],
key: Optional[Callable[[_T], _Sortable]] = ...) -> List[_T]: ...

8
stdlib/2and3/_heapq.pyi
View File

@@ -1,6 +1,7 @@
"""Stub file for the '_heapq' module."""
from typing import TypeVar, List
from typing import TypeVar, List, Iterable, Any, Callable, Optional
import sys
_T = TypeVar("_T")
@@ -11,5 +12,6 @@ def heappush(heap: List[_T], item: _T) -> None: ...
def heappushpop(heap: List[_T], item: _T) -> _T: ...
def heapreplace(heap: List[_T], item: _T) -> _T:
raise IndexError() # if list is empty
def nlargest(a: int, b: List[_T]) -> List[_T]: ...
def nsmallest(a: int, b: List[_T]) -> List[_T]: ...
if sys.version_info < (3,):
def nlargest(n: int, iterable: Iterable[_T]) -> List[_T]: ...
def nsmallest(n: int, iterable: Iterable[_T]) -> List[_T]: ...