Support mypy typing

Makefile

@@ -2,7 +2,7 @@ init:
 	pip install -r dev-requirements.txt
 
 test:
-	py.test tests/ --doctest-modules multiset.py README.rst
+	py.test tests/ --doctest-modules multiset/multiset.py README.rst
 
 check:
 	pylint multiset

dev-requirements.txt

@@ -4,3 +4,5 @@ coverage>=6.3,<8
 setuptools_scm>=7.0,<9.0
 tox>=2.5,<4.0
 pytest-cov>=4.0,<5.0
+mypy>=1.9,<2
+build>=1.2,<2

make.bat

@@ -11,7 +11,7 @@ goto :eof
 goto :eof
 
 :test
-	py.test tests\ --doctest-modules multiset.py README.rst
+	py.test tests\ --doctest-modules multiset/multiset.py README.rst
 goto :eof
 
 :check

multiset/__init__.py

@@ -0,0 +1 @@
+from .multiset import *

multiset.pyi → multiset/multiset.pyi

@@ -2,6 +2,8 @@
 from typing import (Generic, ItemsView, Iterable, Iterator, KeysView, Mapping, Hashable,
                     MutableMapping, Optional, Set, Type, TypeVar, Union, ValuesView, overload)
 
+from _typeshed import SupportsKeysAndGetItem
+
 T = TypeVar('T')
 TElement = TypeVar('TElement', bound=Hashable)
 OtherType = Union[Iterable[TElement], Mapping[TElement, int]]
@@ -84,7 +86,14 @@ class BaseMultiset(Mapping[TElement, int], Generic[TElement]):
 class Multiset(BaseMultiset[TElement], MutableMapping[TElement, int], Generic[TElement]):
     def __setitem__(self, element: TElement, multiplicity: int) -> None: ...
     def __delitem__(self, element: TElement) -> None: ...
+    @overload
     def update(self, *others: OtherType) -> None: ...
+    @overload
+    def update(self, __m: SupportsKeysAndGetItem[T, int], /, **kwargs: int) -> None: ...
+    @overload
+    def update(self, __m: Iterable[tuple[T, int]], /, **kwargs: int) -> None: ...
+    @overload
+    def update(self, **kwargs: int) -> None: ...
     def union_update(self, *others: OtherType) -> None: ...
     @overload
     def __ior__(self: Self, other: Set[TElement]) -> Self: ...

setup.cfg

@@ -29,7 +29,7 @@ setup_requires =
     setuptools >= 46
     setuptools_scm
 python_requires = >= 3.8
-py_modules = multiset
+packages = multiset
 test_suite = tests
 
 [options.package_data]