experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Adding test for BaseResourceVariable.__repr__

Browse Source 
*ding ding* shame shame shame *ding ding*

PiperOrigin-RevId: 302037315
Change-Id: Ifddf2475af96487b522d6b89ae307bf9bf46d5a9
This commit is contained in:
Martin Wicke 2020-03-20 08:59:37 -07:00 committed by TensorFlower Gardener
parent b7855c7098
commit bb97495f77
2 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tensorflow/python/kernel_tests/resource_variable_ops_test.py
View File

@ -208,6 +208,26 @@ class ResourceVariableOpsTest(test_util.TensorFlowTestCase,
resource_variable_ops.assign_variable_op(
handle, constant_op.constant([1.], dtype=dtypes.float32))
def testRepr(self):
with context.eager_mode():
v = resource_variable_ops.ResourceVariable(1)
text = "%r" % v
self.assertEqual(
"<tf.Variable 'Variable:0' shape=() dtype=int32, numpy=1>", text)
def testReprUnavailable(self):
with context.eager_mode():
v = resource_variable_ops.ResourceVariable(1)
# Monkey-patch this variable to not have an available value
def broken_read():
raise ValueError("This doesn't work")
v.read_value = broken_read
text = "%r" % v
self.assertEqual("<tf.Variable 'Variable:0' shape=() dtype=int32,"
" numpy=<unavailable>>", text)
def testUnprintableHandle(self):
with context.eager_mode():
handle = resource_variable_ops.var_handle_op(