Fix test code to not depend on ordering of inputs to addition.

MudeLin · Jul 24, 2015 · 208acb3 · 208acb3
1 parent e7f91a1
commit 208acb3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/theano/sandbox/cuda/tests/test_dnn.py b/theano/sandbox/cuda/tests/test_dnn.py
@@ -700,8 +700,12 @@ def test_dnn_conv_merge_broad():
     fr = conv + lr
 
     f = theano.function([img, kern], [fr])
-    assert not isinstance(f.maker.fgraph.outputs[0].owner.inputs[0].owner.op,
-                          dnn.GpuDnnConv)
+    convs = [n for n in f.maker.fgraph.toposort()
+             if isinstance(n.op, dnn.GpuDnnConv)]
+    assert len(convs) == 1
+    conv = convs[0]
+    # Assert output was not merged
+    assert isinstance(conv.inputs[2].owner.op, GpuAllocEmpty)
 
 
 def test_dnn_conv_grad():