check after comments, too (#13832)

This commit is contained in:
Benjamin Peterson 2012-01-19 17:46:13 -05:00
parent 6a7b855a97
commit cff9237d57
2 changed files with 18 additions and 6 deletions

View File

@ -470,6 +470,8 @@ if 1:
self.assertInvalidSingle('a = 13\nb = 187')
self.assertInvalidSingle('del x\ndel y')
self.assertInvalidSingle('f()\ng()')
self.assertInvalidSingle('f()\n# blah\nblah()')
self.assertInvalidSingle('f()\nxy # blah\nblah()')
def test_main():
support.run_unittest(TestSpecifics)

View File

@ -234,13 +234,23 @@ parsetok(struct tok_state *tok, grammar *g, int start, perrdetail *err_ret,
char *cur = tok->cur;
char c = *tok->cur;
while (c == ' ' || c == '\t' || c == '\n' || c == '\014')
c = *++cur;
for (;;) {
while (c == ' ' || c == '\t' || c == '\n' || c == '\014')
c = *++cur;
if (c && c != '#') {
err_ret->error = E_BADSINGLE;
PyNode_Free(n);
n = NULL;
if (!c)
break;
if (c != '#') {
err_ret->error = E_BADSINGLE;
PyNode_Free(n);
n = NULL;
break;
}
/* Suck up comment. */
while (c && c != '\n')
c = *++cur;
}
}
#endif