Python PythonFile.from_statement 예제들

예제 #1

파일: test_new_style_classes.py 프로젝트: EricCen/commons

def test_new_style_classes():
  nsc = NewStyleClasses(PythonFile.from_statement("""
    class OldStyle:
      pass
    
    class NewStyle(object):
      pass
  """))
  nits = list(nsc.nits())
  assert len(nits) == 1
  assert nits[0]._line_number == 1
  assert nits[0].code == 'T606'
  assert nits[0].severity == Nit.ERROR

  nsc = NewStyleClasses(PythonFile.from_statement("""
    class NewStyle(OtherThing, ThatThing, WhatAmIDoing):
      pass
  """))
  nits = list(nsc.nits())
  assert len(nits) == 0

  nsc = NewStyleClasses(PythonFile.from_statement("""
    class OldStyle():  # unspecified mro
      pass
  """))
  nits = list(nsc.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T606'

예제 #2

파일: test_indentation.py 프로젝트: wcauchois/commons-1

def test_indentation():
  ind = Indentation(PythonFile.from_statement("""
    def foo():
        pass
  """))
  nits = list(ind.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T100'
  assert nits[0].severity == Nit.ERROR

  ind = Indentation(PythonFile.from_statement("""
    def foo():
      pass
  """))
  nits = list(ind.nits())
  assert len(nits) == 0

  ind = Indentation(PythonFile.from_statement("""
    def foo():
      baz = (
          "this "
          "is "
          "ok")
  """))
  nits = list(ind.nits())
  assert len(nits) == 0

예제 #3

def test_new_style_classes():
    nsc = NewStyleClasses(
        PythonFile.from_statement("""
    class OldStyle:
      pass
    
    class NewStyle(object):
      pass
  """))
    nits = list(nsc.nits())
    assert len(nits) == 1
    assert nits[0]._line_number == 1
    assert nits[0].code == 'T606'
    assert nits[0].severity == Nit.ERROR

    nsc = NewStyleClasses(
        PythonFile.from_statement("""
    class NewStyle(OtherThing, ThatThing, WhatAmIDoing):
      pass
  """))
    nits = list(nsc.nits())
    assert len(nits) == 0

    nsc = NewStyleClasses(
        PythonFile.from_statement("""
    class OldStyle():  # unspecified mro
      pass
  """))
    nits = list(nsc.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T606'

예제 #4

파일: test_missing_contextmanager.py 프로젝트: EricCen/commons

def test_missing_contextmanager():
  mcm = MissingContextManager(PythonFile.from_statement("""
    with open("derp.txt"):
      pass
    
    with open("herp.txt") as fp:
      fp.read()
  """))
  nits = list(mcm.nits())
  assert len(nits) == 0

  mcm = MissingContextManager(PythonFile.from_statement("""
    foo = open("derp.txt")
  """))
  nits = list(mcm.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T802'
  assert nits[0].severity == Nit.WARNING

  # TODO(wickman) In these cases suggest using contextlib.closing
  mcm = MissingContextManager(PythonFile.from_statement("""
    from urllib2 import urlopen
    the_googs = urlopen("http://www.google.com").read()
  """))
  nits = list(mcm.nits())
  assert len(nits) == 0

예제 #5

파일: test_print_statements.py 프로젝트: wcauchois/commons-1

def test_print_statements():
    ps = PrintStatements(
        PythonFile.from_statement("""
    from __future__ import print_function
    print("I do what I want")
    
    class Foo(object):
      def print(self):
        "I can do this because it's not a reserved word."
  """))
    assert len(list(ps.nits())) == 0

    ps = PrintStatements(
        PythonFile.from_statement("""
    print("I do what I want")
  """))
    assert len(list(ps.nits())) == 0

    ps = PrintStatements(
        PythonFile.from_statement("""
    print["I do what I want"]
  """))
    nits = list(ps.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T607'
    assert nits[0].severity == Nit.ERROR

예제 #6

def test_missing_contextmanager():
    mcm = MissingContextManager(
        PythonFile.from_statement("""
    with open("derp.txt"):
      pass
    
    with open("herp.txt") as fp:
      fp.read()
  """))
    nits = list(mcm.nits())
    assert len(nits) == 0

    mcm = MissingContextManager(
        PythonFile.from_statement("""
    foo = open("derp.txt")
  """))
    nits = list(mcm.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T802'
    assert nits[0].severity == Nit.WARNING

    # TODO(wickman) In these cases suggest using contextlib.closing
    mcm = MissingContextManager(
        PythonFile.from_statement("""
    from urllib2 import urlopen
    the_googs = urlopen("http://www.google.com").read()
  """))
    nits = list(mcm.nits())
    assert len(nits) == 0

예제 #7

파일: test_newlines.py 프로젝트: EricCen/commons

def test_newlines():
  for toplevel_def in ('def bar():', 'class Bar(object):'):
    for num_newlines in (0, 1, 3, 4):
      newlines = Newlines(PythonFile.from_statement(TOPLEVEL % ('\n' * num_newlines, toplevel_def)))
      nits = list(newlines.nits())
      assert len(nits) == 1
      assert nits[0].code == 'T302'
      assert nits[0].severity == Nit.ERROR
    newlines = Newlines(PythonFile.from_statement(TOPLEVEL % ('\n\n', toplevel_def)))
    assert len(list(newlines.nits())) == 0

예제 #8

파일: test_variable_names.py 프로젝트: wcauchois/commons-1

def test_lower_snake_method_names():
    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    def totally_fine():
      print("Not in a class body")
    
    class DhisRight(object):
      def clearlyNotThinking(self):
        print("In a class body")
  """))
    nits = list(p8.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T002'
    assert nits[0]._line_number == 5
    assert nits[0].severity == Nit.ERROR

    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    class DhisRight:
      def clearlyNotThinking(self):
        print("In a class body")
  """))
    nits = list(p8.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T002'
    assert nits[0]._line_number == 2
    assert nits[0].severity == Nit.ERROR

    # Allow derivations from other modules to be ok.
    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    class TestCase(unittest.TestCase):
      def setUp(self):
        pass
  """))
    nits = list(p8.nits())
    assert len(list(p8.nits())) == 0

    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    def clearlyNotThinking():
      print("Not in a class body")
    
    class DhisRight(object):
      def totally_fine(self):
        print("In a class body")
  """))
    nits = list(p8.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T002'
    assert nits[0]._line_number == 1
    assert nits[0].severity == Nit.ERROR

예제 #9

def test_newlines():
    for toplevel_def in ('def bar():', 'class Bar(object):'):
        for num_newlines in (0, 1, 3, 4):
            newlines = Newlines(
                PythonFile.from_statement(TOPLEVEL %
                                          ('\n' * num_newlines, toplevel_def)))
            nits = list(newlines.nits())
            assert len(nits) == 1
            assert nits[0].code == 'T302'
            assert nits[0].severity == Nit.ERROR
        newlines = Newlines(
            PythonFile.from_statement(TOPLEVEL % ('\n\n', toplevel_def)))
        assert len(list(newlines.nits())) == 0

예제 #10

파일: test_variable_names.py 프로젝트: EricCen/commons

def test_lower_snake_method_names():
  p8 = PEP8VariableNames(PythonFile.from_statement("""
    def totally_fine():
      print("Not in a class body")
    
    class DhisRight(object):
      def clearlyNotThinking(self):
        print("In a class body")
  """))
  nits = list(p8.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T002'
  assert nits[0]._line_number == 5
  assert nits[0].severity == Nit.ERROR

  p8 = PEP8VariableNames(PythonFile.from_statement("""
    class DhisRight:
      def clearlyNotThinking(self):
        print("In a class body")
  """))
  nits = list(p8.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T002'
  assert nits[0]._line_number == 2
  assert nits[0].severity == Nit.ERROR

  # Allow derivations from other modules to be ok.
  p8 = PEP8VariableNames(PythonFile.from_statement("""
    class TestCase(unittest.TestCase):
      def setUp(self):
        pass
  """))
  nits = list(p8.nits())
  assert len(list(p8.nits())) == 0

  p8 = PEP8VariableNames(PythonFile.from_statement("""
    def clearlyNotThinking():
      print("Not in a class body")
    
    class DhisRight(object):
      def totally_fine(self):
        print("In a class body")
  """))
  nits = list(p8.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T002'
  assert nits[0]._line_number == 1
  assert nits[0].severity == Nit.ERROR

예제 #11

파일: test_future_compatibility.py 프로젝트: wcauchois/commons-1

def exemplar_fail(code, severity, statement):
    nits = list(
        FutureCompatibility(PythonFile.from_statement(statement)).nits())
    assert len(nits) == 1
    assert nits[0].code == code
    assert nits[0].severity == severity
    return nits[0]

예제 #12

파일: test_trailing_whitespace.py 프로젝트: wcauchois/commons-1

def test_exception_map():
    tw = TrailingWhitespace(
        PythonFile.from_statement("""
  test_string_001 = ""
  test_string_002 = " "
  test_string_003 = \"\"\"  
    foo   
  \"\"\"
  test_string_006 = ("   "
                     "   ")
  class Foo(object):
    pass
  # comment 010  
  test_string_011 = ''
  # comment 012
  # comment 013
  """))
    assert len(list(tw.nits())) == 0
    assert not tw.has_exception(9, 0, 0)
    assert not tw.has_exception(3, 0, 1)
    assert not tw.has_exception(3, 17, 17)
    assert tw.has_exception(3, 18, 18)
    assert tw.has_exception(3, 18,
                            10000)  # """ continuated strings have no ends.
    assert not tw.has_exception(6, 8, 8)
    assert tw.has_exception(6, 19, 19)
    assert tw.has_exception(6, 19, 23)
    assert not tw.has_exception(
        6, 23, 25)  # ("  " continuations have string termination

예제 #13

파일: test_trailing_whitespace.py 프로젝트: EricCen/commons

def test_exception_map():
  tw = TrailingWhitespace(PythonFile.from_statement("""
  test_string_001 = ""
  test_string_002 = " "
  test_string_003 = \"\"\"  
    foo   
  \"\"\"
  test_string_006 = ("   "
                     "   ")
  class Foo(object):
    pass
  # comment 010  
  test_string_011 = ''
  # comment 012
  # comment 013
  """))
  assert len(list(tw.nits())) == 0
  assert not tw.has_exception(9, 0, 0)
  assert not tw.has_exception(3, 0, 1)
  assert not tw.has_exception(3, 17, 17)
  assert tw.has_exception(3, 18, 18)
  assert tw.has_exception(3, 18, 10000)  # """ continuated strings have no ends. 
  assert not tw.has_exception(6, 8, 8)
  assert tw.has_exception(6, 19, 19)
  assert tw.has_exception(6, 19, 23)
  assert not tw.has_exception(6, 23, 25)  # ("  " continuations have string termination

예제 #14

파일: test_import_order.py 프로젝트: EricCen/commons

def test_import_wildcard():
  io = ImportOrder(PythonFile.from_statement("""
    from twitter.common.dirutil import *
  """))
  nits = list(io.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T400'
  assert nits[0].severity == Nit.ERROR

예제 #15

파일: test_import_order.py 프로젝트: EricCen/commons

def test_import_lexical_order():
  io = ImportOrder(PythonFile.from_statement("""
    from twitter.common.dirutil import safe_rmtree, safe_mkdtemp
  """))
  nits = list(io.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T401'
  assert nits[0].severity == Nit.ERROR

예제 #16

파일: test_trailing_whitespace.py 프로젝트: EricCen/commons

def test_continuation_with_exception():
  tw = TrailingWhitespace(PythonFile.from_statement("""
  test_string_001 = ("   "  
                     "   ")
  """))
  nits = list(tw.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T200'
  assert nits[0].severity == Nit.ERROR

예제 #17

파일: test_noqa.py 프로젝트: EricCen/commons

def test_noqa_file_filter():
  nits = apply_filter(PythonFile.from_statement("""
    # checkstyle: noqa
    print('This is not fine')
    print('This is fine')
  """), Rage, None)
  
  nits = list(nits)
  assert len(nits) == 0

예제 #18

파일: test_noqa.py 프로젝트: EricCen/commons

def test_noqa_line_filter():
  nits = apply_filter(PythonFile.from_statement("""
    print('This is not fine')
    print('This is fine')  # noqa
  """), Rage, None)
  
  nits = list(nits)
  assert len(nits) == 1, ('Actually got nits: %s' % (' '.join('%s:%s' % (nit._line_number, nit) for nit in nits)))
  assert nits[0].code == 'T999'

예제 #19

파일: test_trailing_whitespace.py 프로젝트: wcauchois/commons-1

def test_continuation_with_exception():
    tw = TrailingWhitespace(
        PythonFile.from_statement("""
  test_string_001 = ("   "  
                     "   ")
  """))
    nits = list(tw.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T200'
    assert nits[0].severity == Nit.ERROR

예제 #20

파일: test_variable_names.py 프로젝트: EricCen/commons

def test_class_names():
  p8 = PEP8VariableNames(PythonFile.from_statement("""
    class dhis_not_right(object):
      pass
  """))
  nits = list(p8.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T000'
  assert nits[0]._line_number == 1
  assert nits[0].severity == Nit.ERROR

예제 #21

파일: test_noqa.py 프로젝트: wcauchois/commons-1

def test_noqa_file_filter():
    nits = apply_filter(
        PythonFile.from_statement("""
    # checkstyle: noqa
    print('This is not fine')
    print('This is fine')
  """), Rage, None)

    nits = list(nits)
    assert len(nits) == 0

예제 #22

파일: test_newlines.py 프로젝트: EricCen/commons

def test_classdefs():
  newlines = Newlines(PythonFile.from_statement(GOOD_CLASS_DEF_1))
  assert len(list(newlines.nits())) == 0

  newlines = Newlines(PythonFile.from_statement(GOOD_CLASS_DEF_2))
  assert len(list(newlines.nits())) == 0

  newlines = Newlines(PythonFile.from_statement(BAD_CLASS_DEF_1))
  nits = list(newlines.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T301'
  assert nits[0]._line_number == 4
  assert nits[0].severity == Nit.ERROR

  newlines = Newlines(PythonFile.from_statement(BAD_CLASS_DEF_2))
  nits = list(newlines.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T301'
  assert nits[0]._line_number == 7
  assert nits[0].severity == Nit.ERROR

예제 #23

파일: test_variable_names.py 프로젝트: EricCen/commons

def test_class_globals():
  p8 = PEP8VariableNames(PythonFile.from_statement("""
    class DhisRight(object):
      RIGHT = 123
      notRight = 321
  """))
  nits = list(p8.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T001'
  assert nits[0]._line_number == 3
  assert nits[0].severity == Nit.ERROR

예제 #24

파일: test_variable_names.py 프로젝트: wcauchois/commons-1

def test_class_names():
    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    class dhis_not_right(object):
      pass
  """))
    nits = list(p8.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T000'
    assert nits[0]._line_number == 1
    assert nits[0].severity == Nit.ERROR

예제 #25

def test_classdefs():
    newlines = Newlines(PythonFile.from_statement(GOOD_CLASS_DEF_1))
    assert len(list(newlines.nits())) == 0

    newlines = Newlines(PythonFile.from_statement(GOOD_CLASS_DEF_2))
    assert len(list(newlines.nits())) == 0

    newlines = Newlines(PythonFile.from_statement(BAD_CLASS_DEF_1))
    nits = list(newlines.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T301'
    assert nits[0]._line_number == 4
    assert nits[0].severity == Nit.ERROR

    newlines = Newlines(PythonFile.from_statement(BAD_CLASS_DEF_2))
    nits = list(newlines.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T301'
    assert nits[0]._line_number == 7
    assert nits[0].severity == Nit.ERROR

예제 #26

파일: test_noqa.py 프로젝트: wcauchois/commons-1

def test_noqa_line_filter():
    nits = apply_filter(
        PythonFile.from_statement("""
    print('This is not fine')
    print('This is fine')  # noqa
  """), Rage, None)

    nits = list(nits)
    assert len(nits) == 1, ('Actually got nits: %s' %
                            (' '.join('%s:%s' % (nit._line_number, nit)
                                      for nit in nits)))
    assert nits[0].code == 'T999'

예제 #27

파일: test_variable_names.py 프로젝트: wcauchois/commons-1

def test_class_globals():
    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    class DhisRight(object):
      RIGHT = 123
      notRight = 321
  """))
    nits = list(p8.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T001'
    assert nits[0]._line_number == 3
    assert nits[0].severity == Nit.ERROR

예제 #28

파일: test_print_statements.py 프로젝트: EricCen/commons

def test_print_statements():
  ps = PrintStatements(PythonFile.from_statement("""
    from __future__ import print_function
    print("I do what I want")
    
    class Foo(object):
      def print(self):
        "I can do this because it's not a reserved word."
  """))
  assert len(list(ps.nits())) == 0

  ps = PrintStatements(PythonFile.from_statement("""
    print("I do what I want")
  """))
  assert len(list(ps.nits())) == 0

  ps = PrintStatements(PythonFile.from_statement("""
    print["I do what I want"]
  """))
  nits = list(ps.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T607'
  assert nits[0].severity == Nit.ERROR

예제 #29

파일: test_variable_names.py 프로젝트: EricCen/commons

def test_builtin_overrides():
  p8 = PEP8VariableNames(PythonFile.from_statement("""
    def range():
      print("Not in a class body")
    
    class DhisRight(object):
      def any(self):
        print("In a class body")
  """))
  nits = list(p8.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T801'
  assert nits[0]._line_number == 1
  assert nits[0].severity == Nit.ERROR

예제 #30

파일: test_trailing_whitespace.py 프로젝트: EricCen/commons

def test_trailing_slash():
  tw = TrailingWhitespace(PythonFile.from_statement("""
  foo = \\
    123
  bar = \"\"\"
    bin/bash foo \\
             bar \\
             baz
  \"\"\"
  """))
  nits = list(tw.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T201'
  assert nits[0].severity == Nit.ERROR
  assert nits[0]._line_number == 1

예제 #31

파일: test_variable_names.py 프로젝트: wcauchois/commons-1

def test_builtin_overrides():
    p8 = PEP8VariableNames(
        PythonFile.from_statement("""
    def range():
      print("Not in a class body")
    
    class DhisRight(object):
      def any(self):
        print("In a class body")
  """))
    nits = list(p8.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T801'
    assert nits[0]._line_number == 1
    assert nits[0].severity == Nit.ERROR

예제 #32

파일: test_trailing_whitespace.py 프로젝트: wcauchois/commons-1

def test_trailing_slash():
    tw = TrailingWhitespace(
        PythonFile.from_statement("""
  foo = \\
    123
  bar = \"\"\"
    bin/bash foo \\
             bar \\
             baz
  \"\"\"
  """))
    nits = list(tw.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T201'
    assert nits[0].severity == Nit.ERROR
    assert nits[0]._line_number == 1

예제 #33

파일: test_future_compatibility.py 프로젝트: EricCen/commons

def exemplar_fail(code, severity, statement):
    nits = list(FutureCompatibility(PythonFile.from_statement(statement)).nits())
    assert len(nits) == 1
    assert nits[0].code == code
    assert nits[0].severity == severity
    return nits[0]

예제 #34

파일: test_class_factoring.py 프로젝트: EricCen/commons

from twitter.checkstyle.common import Nit, PythonFile
from twitter.checkstyle.plugins.class_factoring import ClassFactoring


BAD_CLASS = PythonFile.from_statement("""
class Distiller(object):
  CONSTANT = "foo"

  def foo(self, value):
    return os.path.join(Distiller.CONSTANT, value)
""")


def test_class_factoring():
  plugin = ClassFactoring(BAD_CLASS)
  nits = list(plugin.nits())
  assert len(nits) == 1
  assert nits[0].code == 'T800'
  assert nits[0].severity == Nit.WARNING

예제 #35

파일: test_class_factoring.py 프로젝트: wcauchois/commons-1

from twitter.checkstyle.common import Nit, PythonFile
from twitter.checkstyle.plugins.class_factoring import ClassFactoring

BAD_CLASS = PythonFile.from_statement("""
class Distiller(object):
  CONSTANT = "foo"

  def foo(self, value):
    return os.path.join(Distiller.CONSTANT, value)
""")


def test_class_factoring():
    plugin = ClassFactoring(BAD_CLASS)
    nits = list(plugin.nits())
    assert len(nits) == 1
    assert nits[0].code == 'T800'
    assert nits[0].severity == Nit.WARNING

예제 #36

파일: test_except_statements.py 프로젝트: wcauchois/commons-1

def nits_from(clause):
  return list(ExceptStatements(PythonFile.from_statement(EXCEPT_TEMPLATE % clause)).nits())

예제 #37

파일: test_future_compatibility.py 프로젝트: EricCen/commons

def exemplar_pass(statement):
    nits = list(FutureCompatibility(PythonFile.from_statement(statement)).nits())
    assert len(nits) == 0

예제 #38

파일: test_future_compatibility.py 프로젝트: wcauchois/commons-1

def exemplar_pass(statement):
    nits = list(
        FutureCompatibility(PythonFile.from_statement(statement)).nits())
    assert len(nits) == 0