def tokeneater(self, type, token, srowcol, erowcol, line):
if not self.started and not self.indecorator:
- if type == tokenize.INDENT or token == "async":
+ if type in (tokenize.INDENT, tokenize.COMMENT, tokenize.NL):
+ pass
+ elif token == "async":
pass
# skip any decorators
elif token == "@":
self.assertSourceEqual(next(mod2.ge377), 377, 380)
self.assertSourceEqual(next(mod2.func383()), 385, 388)
+ def test_comment_or_empty_line_after_decorator(self):
+ self.assertSourceEqual(mod2.func394, 392, 395)
+ self.assertSourceEqual(mod2.func400, 398, 401)
+
class TestNoEOL(GetSourceBase):
def setUp(self):