Fix running test_tokenize directly

diff --git a/Lib/test/test_tokenize.py b/Lib/test/test_tokenize.py
index 90438e7d3077d7d0208b6735a0bce87e9717e1f5..77c0423d19e8818b09417da2e17ef4a1b09f9783 100644 (file)
--- a/Lib/test/test_tokenize.py
+++ b/Lib/test/test_tokenize.py
@@ -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()