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

TF Framework ops_eager_test missing test case add

Browse Source 
1-disable_eager_execution api test case
This commit is contained in:
Dayananda-V 2019-03-06 11:24:53 +05:30
parent 91950501a4
commit 9bb19884c3
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/python/framework/ops_enable_eager_test.py
View File

@ -23,9 +23,10 @@ from tensorflow.python.framework import ops
from tensorflow.python.platform import googletest
class OpsEnableEagerTest(googletest.TestCase):
class OpsEnableAndDisableEagerTest(googletest.TestCase):
def test_enable_eager_execution_multiple_times(self):
def setUp(self):
# test for enable eager test
ops.enable_eager_execution()
self.assertTrue(context.executing_eagerly())
@ -33,6 +34,15 @@ class OpsEnableEagerTest(googletest.TestCase):
ops.enable_eager_execution()
self.assertTrue(context.executing_eagerly())
def tearDown(self):
# test for disable eager test
ops.disable_eager_execution()
self.assertFalse(context.executing_eagerly())
# Calling disable eager execution a second time should not cause an error.
ops.disable_eager_execution()
self.assertFalse(context.executing_eagerly())
if __name__ == '__main__':
googletest.main()