@@ -802,14 +802,8 @@ def test_class_base_props(self) -> None:
802
802
def test_class_locals (self ) -> None :
803
803
"""Test the 'locals' dictionary of an astroid class."""
804
804
module = self .module
805
- klass1 = module ["YO" ]
806
- locals1 = klass1 .locals
807
- keys = sorted (locals1 .keys ())
808
805
assert_keys = ["__annotations__" , "__init__" , "__module__" , "__qualname__" , "a" ]
809
- self .assertEqual (keys , assert_keys )
810
- klass2 = module ["YOUPI" ]
811
- locals2 = klass2 .locals
812
- keys = locals2 .keys ()
806
+ self .assertEqual (sorted (module ["YO" ].locals .keys ()), assert_keys )
813
807
assert_keys = [
814
808
"__annotations__" ,
815
809
"__init__" ,
@@ -820,39 +814,35 @@ def test_class_locals(self) -> None:
820
814
"method" ,
821
815
"static_method" ,
822
816
]
823
- self .assertEqual (sorted (keys ), assert_keys )
817
+ self .assertEqual (sorted (module [ "YOUPI" ]. locals . keys () ), assert_keys )
824
818
825
819
def test_class_instance_attrs (self ) -> None :
826
820
module = self .module
827
- klass1 = module ["YO" ]
828
- klass2 = module ["YOUPI" ]
829
- self .assertEqual (list (klass1 .instance_attrs .keys ()), ["yo" ])
830
- self .assertEqual (list (klass2 .instance_attrs .keys ()), ["member" ])
821
+ self .assertEqual (list (module ["YO" ].instance_attrs .keys ()), ["yo" ])
822
+ self .assertEqual (list (module ["YOUPI" ].instance_attrs .keys ()), ["member" ])
831
823
832
824
def test_class_basenames (self ) -> None :
833
825
module = self .module
834
- klass1 = module ["YO" ]
835
- klass2 = module ["YOUPI" ]
836
- self .assertEqual (klass1 .basenames , [])
837
- self .assertEqual (klass2 .basenames , ["YO" ])
826
+ self .assertEqual (module ["YO" ].basenames , [])
827
+ self .assertEqual (module ["YOUPI" ].basenames , ["YO" ])
838
828
839
829
def test_method_base_props (self ) -> None :
840
830
"""Test base properties and method of an astroid method."""
841
- klass2 = self .module ["YOUPI" ]
842
- # "normal" method
843
- method = klass2 ["method" ]
831
+ method = self .module ["YOUPI" ]["method" ]
844
832
self .assertEqual (method .name , "method" )
845
833
self .assertEqual ([n .name for n in method .args .args ], ["self" ])
846
834
assert isinstance (method .doc_node , nodes .Const )
847
835
self .assertEqual (method .doc_node .value , "method\n test" )
848
836
self .assertEqual (method .fromlineno , 48 )
849
837
self .assertEqual (method .type , "method" )
850
- # class method
851
- method = klass2 ["class_method" ]
838
+
839
+ def test_class_method_base_props (self ) -> None :
840
+ method = self .module ["YOUPI" ]["class_method" ]
852
841
self .assertEqual ([n .name for n in method .args .args ], ["cls" ])
853
842
self .assertEqual (method .type , "classmethod" )
854
- # static method
855
- method = klass2 ["static_method" ]
843
+
844
+ def test_static_method_base_props (self ) -> None :
845
+ method = self .module ["YOUPI" ]["static_method" ]
856
846
self .assertEqual (method .args .args , [])
857
847
self .assertEqual (method .type , "staticmethod" )
858
848
0 commit comments