fixed indentation
This commit is contained in:
parent
6be0e40016
commit
0fb2791e0f
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user