frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-22 06:11:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix all the broken tests from adding debug info (#2013)

Browse Source
This commit is contained in:
Bram Wasti
2018-02-22 17:43:53 -08:00
committed by bddppq
parent 38f18c1daa
commit 51897e52da
2 changed files with 34 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
caffe2/python/core_test.py
View File

@ -259,6 +259,12 @@ class TestCreateOperator(test_util.TestCase):
class TestAutoNaming(test_util.TestCase):
def assertOperatorListEqual(self, operatorDefList1, operatorDefList2):
for op in operatorDefList1:
op.debug_info = ""
for op in operatorDefList2:
op.debug_info = ""
self.assertEqual(operatorDefList1, operatorDefList2)
"""
Test that operators are named with different names, and that automatically
named blob names don't clash intra or inter networks.
@ -275,7 +281,7 @@ class TestAutoNaming(test_util.TestCase):
net_a = create_net()
net_b = create_net()
# created net proto is predicatable.
self.assertEqual(net_a.Proto().op,
self.assertOperatorListEqual(net_a.Proto().op,
net_b.Proto().op)
self.assertEqual(net_a.Proto().op[0].output[0], 'foo/ab')
self.assertEqual(net_a.Proto().op[1].output[0], 'cd')