Ejemplo n.º 1
0
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'
Ejemplo n.º 2
0
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'