def test_get_class_name_simple(): code = [ 'class TheName(object):', ' a = 1', ' b = 2', ] lineno = 3 # b = 2 assert get_class_name(code, lineno) == 'TheName'
def test_get_outer_class_from_within_method(): code = [ 'class Outer(object):', ' class Inner(object):', ' a = 2', ' def method():', ' c = 5' ] lineno = 5 # c = 5 result = get_class_name(code, lineno) assert result == 'Outer'