From 300f8e54ca006b3c504e0708b7234ccb01d22012 Mon Sep 17 00:00:00 2001 From: Hajime Senuma Date: Tue, 2 Jan 2024 17:57:58 +0900 Subject: [PATCH] Fix type hints --- src/mmh3/__init__.pyi | 6 +++--- tests/test_mmh3.py | 5 +++++ 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/src/mmh3/__init__.pyi b/src/mmh3/__init__.pyi index faf8804..fa189bd 100644 --- a/src/mmh3/__init__.pyi +++ b/src/mmh3/__init__.pyi @@ -11,9 +11,9 @@ StrHashable = Union[str, Hashable] def hash(key: StrHashable, seed: int = 0, signed: bool = True) -> int: ... def hash_from_buffer(key: StrHashable, seed: int = 0, signed: bool = True) -> int: ... -def hash64(key: StrHashable, seed: int = 0, signed: bool = True) -> tuple[int, int]: ... -def hash128(key: StrHashable, seed: int = 0, signed: bool = True) -> int: ... -def hash_bytes(key: StrHashable, seed: int = 0, signed: bool = True) -> bytes: ... +def hash64(key: StrHashable, seed: int = 0, x64arch:bool = True, signed: bool = True) -> tuple[int, int]: ... +def hash128(key: StrHashable, seed: int = 0, x64arch:bool = True, signed: bool = False) -> int: ... +def hash_bytes(key: StrHashable, seed: int = 0, x64arch: bool = True) -> bytes: ... class Hasher: def __init__(self, seed: int = 0) -> None: ... diff --git a/tests/test_mmh3.py b/tests/test_mmh3.py index a8fead9..0eee15d 100644 --- a/tests/test_mmh3.py +++ b/tests/test_mmh3.py @@ -169,6 +169,11 @@ def test_hash_from_buffer() -> None: def test_hash_bytes() -> None: assert mmh3.hash_bytes("foo") == b"aE\xf5\x01W\x86q\xe2\x87}\xba+\xe4\x87\xaf~" + assert ( + mmh3.hash_bytes("foo", 0, True) + == b"aE\xf5\x01W\x86q\xe2\x87}\xba+\xe4\x87\xaf~" + ) + assert mmh3.hash_bytes("foo", 0, x64arch=False) == b"%\x1b|We%\xb6`e%\xb6`e%\xb6`" # TODO