diff --git a/tensorflow/python/framework/sparse_tensor_test.py b/tensorflow/python/framework/sparse_tensor_test.py index 736543f669b..aee0894d85d 100644 --- a/tensorflow/python/framework/sparse_tensor_test.py +++ b/tensorflow/python/framework/sparse_tensor_test.py @@ -99,16 +99,16 @@ class SparseTensorTest(test_util.TensorFlowTestCase): self.assertIn(out.op, sp.consumers()) def testWithValues(self): - source = sparse_tensor.SparseTensor( - indices=[[0, 0], [1, 2]], values=[1., 2], dense_shape=[3, 4]) - new_tensor = source.with_values([5.0, 1.0]) - self.assertAllEqual(new_tensor.indices, source.indices) - self.assertAllEqual(new_tensor.values, [5.0, 1.0]) - self.assertAllEqual(new_tensor.dense_shape, source.dense_shape) + source = sparse_tensor.SparseTensor( + indices=[[0, 0], [1, 2]], values=[1., 2], dense_shape=[3, 4]) + new_tensor = source.with_values([5.0, 1.0]) + self.assertAllEqual(new_tensor.indices, source.indices) + self.assertAllEqual(new_tensor.values, [5.0, 1.0]) + self.assertAllEqual(new_tensor.dense_shape, source.dense_shape) - # ensure new value's shape is checked - with self.assertRaises((errors.InvalidArgumentError, ValueError)): - source.with_values([[5.0, 1.0]]) + # ensure new value's shape is checked + with self.assertRaises((errors.InvalidArgumentError, ValueError)): + source.with_values([[5.0, 1.0]]) class ConvertToTensorOrSparseTensorTest(test_util.TensorFlowTestCase):