1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#!/usr/bin/env python3
import copy
from collections.abc import Hashable
from typing import ItemsView, Iterator, KeysView, Mapping, ValuesView
from frozendict import frozendict
from typing_extensions import assert_type
fd = frozendict(a=1, b=2)
assert_type(frozendict(), frozendict[()])
assert_type(fd, frozendict[str, int])
assert_type(fd["a"], int)
assert_type(iter(fd), Iterator[str])
assert_type(frozendict.fromkeys("abc", 0), frozendict[str, int])
assert_type(fd.copy(), frozendict[str, int])
assert_type(fd.item(0), tuple[str, int])
assert_type(fd.item(), tuple[str, int])
assert_type(fd.key(0), str)
assert_type(fd.key(), str)
assert_type(fd.value(0), int)
assert_type(fd.value(), int)
assert_type(fd.get("a"), int | None)
assert_type(fd.items(), ItemsView[str, int])
assert_type(fd.keys(), KeysView[str])
assert_type(fd.values(), ValuesView[int])
assert_type(copy.copy(fd), frozendict[str, int])
assert_type(copy.deepcopy(fd), frozendict[str, int])
assert_type(reversed(fd), reversed)
assert_type(fd.delete("a"), frozendict[str, int])
assert_type(fd | {"c": 2}, frozendict[str, int])
assert_type(fd | {1: 2}, frozendict[str | int, int])
assert_type(fd | {"c": "c"}, frozendict[str, str | int])
assert_type(fd | {1: "c"}, frozendict[str | int, str | int])
assert_type(fd.setdefault("a", 0), frozendict[str, int])
assert_type(fd.setdefault("a", "a"), frozendict[str, str | int])
assert_type(fd.setdefault(0, 0), frozendict[str | int, int])
assert_type(fd.setdefault(0, "a"), frozendict[str | int, str | int])
assert_type(fd.set("abc", 1), frozendict[str, int])
assert_type(fd.set("abc", "abc"), frozendict[str, str | int])
assert_type(fd.set(1, 1), frozendict[str | int, int])
assert_type(fd.set(1, "abc"), frozendict[str | int, str | int])
# frozendict implements Mapping[K, V] and is covariant in V
vals: frozendict[str, Hashable] = fd
mapping: Mapping[str, Hashable] = fd
|
