def test_blank_lines_missing_newline_windows_fix(): rules = [BlankLinesRule()] src = "def test():\r\n pass\r\n\r\nclass Two:\r\n pass\r\n" expected = "def test():\r\n pass\r\n\r\n\r\nclass Two:\r\n pass\r\n" actual = linter(src, rules, fix=True) assert src_equal(actual, expected)
def test_blank_lines_internal_decorator_fix(): rules = [BlankLinesRule()] src = """ def test(): pass class Test(object): def one(): pass @a_decorator def wow(): pass """ expected = """ def test(): pass class Test(object): def one(): pass @a_decorator def wow(): pass """ actual = linter(src, rules, fix=True) assert src_equal(actual, expected)
def test_comments_separating_funcs_pass(): rules = [BlankLinesRule({'rule_settings': "error"})] src = """ def remove_farewell(text): text = "" return text # def word_tokenize(text, nlp): # doc = nlp.tokenizer(text) # tokens = [word.string.strip() for word in doc] # return tokens # def pos_tag(text, nlp): # doc = nlp(text) # tags = [(word.string.strip(), word.tag_) for word in doc] # return(tags) def word_tokenize(text): return wrd_tokenize(text) """ expected = [] actual = linter(src, rules, fix=False) assert actual['errors'] == expected
def test_blank_lines_internal_func_fix(): rules = [BlankLinesRule()] src = """ def test(): pass class Two(object): def cool(): pass def wow(): pass """ expected = """ def test(): pass class Two(object): def cool(): pass def wow(): pass """ actual = linter(src, rules, fix=True) assert src_equal(actual, expected)
def test_blank_lines_w_comments_fix(): rules = [BlankLinesRule()] src = """ class Test(object): pass # one # three # two class TestTwo(object): pass """ expected = """ class Test(object): pass # one # three # two class TestTwo(object): pass """ actual = linter(src, rules, fix=True) assert src_equal(actual, expected)
def test_method_decorated_pass(): rules = [BlankLinesRule({'rule_settings': "error"})] src = """ class Vocab(object): def __init__(self): pass @staticmethod def boo(): pass """ expected = [] actual = linter(src, rules, fix=False) assert actual['errors'] == expected
def test_nested_indent_pass(): rules = [BlankLinesRule({'rule_settings': "error"})] src = """ def write_TSV(filename, features, labels): with open(filename, "w") as f: f.write(header) def main(): # Get credentials and set up acces to Gmail API pass """ expected = [] actual = linter(src, rules, fix=False) assert actual['errors'] == expected
def test_blank_lines_missing_newline_error(): rules = [BlankLinesRule({'rule_settings': "error"})] src = """ def test(): pass class Two(object): pass """ expected = [{ "name": "blank_lines", "filename": "", "lineno": 5, "msg": "expected 2 blank lines, found 1", }] actual = linter(src, rules, fix=False) assert actual['errors'] == expected
def test_blank_lines_missing_multiple_newlines_fix(): rules = [BlankLinesRule()] src = """ def test(): pass class Two(object): pass """ expected = """ def test(): pass class Two(object): pass """ actual = linter(src, rules, fix=True) assert src_equal(actual, expected)
def test_blank_lines_setting_four_fix(): rules = [BlankLinesRule({'num_newlines': 4})] src = """ def test(): pass class Two(object): pass """ expected = """ def test(): pass class Two(object): pass """ actual = linter(src, rules, fix=True) assert src_equal(actual, expected)