Fix running test_tokenize directly

This commit is contained in:
Zachary Ware 2016-09-09 12:55:14 -07:00
parent c1bd632abf
commit a0154c0f0e
1 changed files with 2 additions and 2 deletions

View File

@ -3,7 +3,7 @@ from tokenize import (tokenize, _tokenize, untokenize, NUMBER, NAME, OP,
STRING, ENDMARKER, ENCODING, tok_name, detect_encoding,
open as tokenize_open, Untokenizer)
from io import BytesIO
from unittest import TestCase, mock
from unittest import TestCase, mock, main
import os
import token
@ -1564,4 +1564,4 @@ class TestRoundtrip(TestCase):
if __name__ == "__main__":
unittest.main()
main()