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

Grappler item_test: fix item order dependency

Browse Source 
by replacing assertEqual() with assertItemsEqual()

Fixes test failures such as:
http://ci.tensorflow.org/view/Tensorflow%20Jenkins%20Monitored%20builds/job/nightly-matrix-cpu/TF_BUILD_IS_PIP=NO_PIP,TF_BUILD_PYTHON_VERSION=PYTHON3,label=cpu-slave/763/consoleFull

PiperOrigin-RevId: 180550331
This commit is contained in:
Shanqing Cai 2018-01-02 08:38:44 -08:00 committed by TensorFlower Gardener
parent e415e562d4
commit dfac1ba6e6
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/grappler/item_test.py
View File

@ -56,7 +56,7 @@ class ItemTest(test.TestCase):
mg = meta_graph.create_meta_graph_def(graph=g)
grappler_item = item.Item(mg)
op_list = grappler_item.IdentifyImportantOps()
self.assertEqual([b'Const', b'Const_1', b'add'], op_list)
self.assertItemsEqual([b'Const', b'Const_1', b'add'], op_list)
def testOpProperties(self):
with ops.Graph().as_default() as g:
@ -119,9 +119,8 @@ class ItemTest(test.TestCase):
grappler_item = item.Item(mg)
groups = grappler_item.GetColocationGroups()
self.assertEqual(len(groups), 1)
self.assertEqual(
sorted(groups[0]),
['Assign', 'RefIdentity', 'Variable', 'Variable/Assign'])
self.assertItemsEqual(
groups[0], ['Assign', 'RefIdentity', 'Variable', 'Variable/Assign'])
if __name__ == '__main__':