Merge pull request #1172 from rizar/fix_test_error_message

Fix tests for the convolutional brick
mila-iqia · Jan 4, 2017 · b22979b · b22979b
2 parents 3db89c0 + 08c00d9
commit b22979b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/bricks/test_conv.py b/tests/bricks/test_conv.py
@@ -214,19 +214,19 @@ def test_untied_biases():
 
     # Untied biases require images of a specific size
     x_val_1 = numpy.ones((batch_size, num_channels, 28, 30),
-                          dtype=theano.config.floatX)
+                         dtype=theano.config.floatX)
 
     assert_allclose(func(x_val_1),
                     numpy.prod(filter_size) * num_channels *
                     numpy.ones((batch_size, num_filters, 26, 28)) + 2)
 
     x_val_2 = numpy.ones((batch_size, num_channels, 23, 19),
-                          dtype=theano.config.floatX)
+                         dtype=theano.config.floatX)
 
     def wrongsize():
         func(x_val_2)
 
-    assert_raises_regexp(ValueError, 'Input dimension mis-match.*',
+    assert_raises_regexp(AssertionError, 'AbstractConv shape mismatch',
                          wrongsize)