@@ -2918,10 +2918,18 @@ def wrapper(*args, **kwargs):
29182918 return _ (* args , ** kwargs )
29192919
29202920 class SomeClass :
2921- def method (self , arg : dict ):
2921+ def for_dict (self , arg : dict ):
29222922 return "dict"
29232923
2924- t .register (SomeClass ().method )
2924+ def for_set (self , arg : set , arg2 : None ):
2925+ return "set"
2926+
2927+ def for_complex (self : object , arg : complex , arg2 : None ):
2928+ return "complex"
2929+
2930+ t .register (SomeClass ().for_dict )
2931+ t .register (SomeClass ().for_set )
2932+ t .register (SomeClass ().for_complex )
29252933
29262934 self .assertEqual (t (0 ), "int" )
29272935 self .assertEqual (t ('' ), "str" )
@@ -2930,6 +2938,8 @@ def method(self, arg: dict):
29302938 self .assertEqual (t (NotImplemented ), "base" )
29312939 self .assertEqual (t (b'' ), "bytes" )
29322940 self .assertEqual (t ({}), "dict" )
2941+ self .assertEqual (t (set (), None ), "set" )
2942+ self .assertEqual (t (0j , None ), "complex" )
29332943
29342944 def test_method_type_ann_register (self ):
29352945
0 commit comments