clean up tests

tests/optimization_test.py

@@ -20,7 +20,7 @@ import unittest
 
 import torch
 
-import optimization as optimization
+import optimization
 
 class OptimizationTest(unittest.TestCase):
 

tests/tokenization_test.py

@@ -19,7 +19,7 @@ from __future__ import print_function
 import os
 import unittest
 
-import tokenization as tokenization
+import tokenization
 
 
 class TokenizationTest(unittest.TestCase):