@@ -55,28 +55,28 @@ def test_build_class(self) -> None:
55
55
self .assertEqual (node .doc_node , None )
56
56
57
57
def test_build_function (self ) -> None :
58
- node = build_function ("MyFunction" )
58
+ node = build_function ("MyFunction" , DUMMY_MOD )
59
59
self .assertEqual (node .name , "MyFunction" )
60
60
self .assertEqual (node .doc_node , None )
61
61
62
62
def test_build_function_args (self ) -> None :
63
63
args = ["myArgs1" , "myArgs2" ]
64
- node = build_function ("MyFunction" , args )
64
+ node = build_function ("MyFunction" , DUMMY_MOD , args )
65
65
self .assertEqual ("myArgs1" , node .args .args [0 ].name )
66
66
self .assertEqual ("myArgs2" , node .args .args [1 ].name )
67
67
self .assertEqual (2 , len (node .args .args ))
68
68
69
69
def test_build_function_defaults (self ) -> None :
70
70
defaults = ["defaults1" , "defaults2" ]
71
- node = build_function (name = "MyFunction" , args = None , defaults = defaults )
71
+ node = build_function ("MyFunction" , DUMMY_MOD , args = None , defaults = defaults )
72
72
self .assertEqual (2 , len (node .args .defaults ))
73
73
74
74
def test_build_function_posonlyargs (self ) -> None :
75
- node = build_function (name = "MyFunction" , posonlyargs = ["a" , "b" ])
75
+ node = build_function ("MyFunction" , DUMMY_MOD , posonlyargs = ["a" , "b" ])
76
76
self .assertEqual (2 , len (node .args .posonlyargs ))
77
77
78
78
def test_build_function_kwonlyargs (self ) -> None :
79
- node = build_function (name = "MyFunction" , kwonlyargs = ["a" , "b" ])
79
+ node = build_function ("MyFunction" , DUMMY_MOD , kwonlyargs = ["a" , "b" ])
80
80
assert len (node .args .kwonlyargs ) == 2
81
81
assert node .args .kwonlyargs [0 ].name == "a"
82
82
assert node .args .kwonlyargs [1 ].name == "b"
0 commit comments