Remove unittest import (#11596)

This commit is contained in:
Jonathan Hseu 2017-07-18 19:40:38 -07:00 committed by Frank Chen
parent 53d4c4b80f
commit daa67ad176

View File

@ -19,7 +19,6 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import numpy as np import numpy as np
import unittest
from tensorflow.python.framework import constant_op from tensorflow.python.framework import constant_op
from tensorflow.python.framework import dtypes from tensorflow.python.framework import dtypes
@ -606,7 +605,6 @@ class SparseReduceTest(test_util.TensorFlowTestCase):
self._compare(sp_t, reduction_axes, ndims, True, False) self._compare(sp_t, reduction_axes, ndims, True, False)
self._compare(sp_t, reduction_axes, ndims, True, True) self._compare(sp_t, reduction_axes, ndims, True, True)
@unittest.skipIf(np.__version__ == "1.13.0", "numpy 1.13 bug")
def testSimpleAndRandomInputs(self): def testSimpleAndRandomInputs(self):
if np.__version__ == "1.13.0": if np.__version__ == "1.13.0":
self.skipTest("numpy 1.13.0 bug") self.skipTest("numpy 1.13.0 bug")
@ -646,7 +644,6 @@ class SparseReduceTest(test_util.TensorFlowTestCase):
with self.assertRaisesOpError("Invalid reduction dimension 2"): with self.assertRaisesOpError("Invalid reduction dimension 2"):
sparse_ops.sparse_reduce_max(sp_t, 2).eval() sparse_ops.sparse_reduce_max(sp_t, 2).eval()
@unittest.skipIf(np.__version__ == "1.13.0", "numpy 1.13 bug")
def testGradient(self): def testGradient(self):
if np.__version__ == "1.13.0": if np.__version__ == "1.13.0":
self.skipTest("numpy 1.13.0 bug") self.skipTest("numpy 1.13.0 bug")