re-add kwargs to __new__ (#273)

corenting · Oct 16, 2023 · bdf02ce · bdf02ce
1 parent 7918f3b
commit bdf02ce
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/immutabledict/__init__.py b/immutabledict/__init__.py
@@ -39,9 +39,9 @@ def fromkeys(
     ) -> immutabledict[_K, _V]:
         return cls(cls.dict_cls.fromkeys(seq, value))
 
-    def __new__(cls, *args: Any) -> immutabledict[_K, _V]:
+    def __new__(cls, *args: Any, **kwargs: Any) -> immutabledict[_K, _V]:
         inst = super().__new__(cls)
-        setattr(inst, "_dict", cls.dict_cls(*args))
+        setattr(inst, "_dict", cls.dict_cls(*args, **kwargs))
         setattr(inst, "_hash", None)
         return inst
 

diff --git a/tests/test_immutabledict.py b/tests/test_immutabledict.py
@@ -215,6 +215,10 @@ def test_performance(self, statement: str) -> None:
 
         assert time_immutable < 1.2 * time_standard
 
+    def test_new_kwargs(self) -> None:
+        immutable_dict: immutabledict[str, int] = immutabledict(a=1, b=2)
+        assert immutable_dict == {"a": 1, "b": 2} == dict(a=1, b=2)
+
 
 class TestImmutableOrderedDict:
     def test_ordered(self) -> None: