Fixed tests of check_grads in Python 3 (differences in function strin…

…g representation between Python 2 and 3)
HIPS · Sep 6, 2017 · 4e517a0 · 4e517a0
1 parent dcf4f7d
commit 4e517a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/autograd/test_util.py b/autograd/test_util.py
@@ -34,7 +34,7 @@ def check_vjp(f, x):
     assert scalar_close(vjv_numeric, vjv_exact), \
         ("Derivative (VJP) check of {} failed with arg {}:\n"
          "analytic: {}\nnumeric:  {}".format(
-            f, x, vjv_numeric, vjv_exact))
+            get_name(f), x, vjv_numeric, vjv_exact))
 
 def check_jvp(f, x):
     jvp = make_jvp(f, x)

diff --git a/tests/test_tests.py b/tests/test_tests.py
@@ -10,7 +10,7 @@ def foo(x):
     defvjp(foo, lambda ans, vs, gvs, x : lambda g: g * 2.001)
 
     assert_raises_regexp(AssertionError,
-                         "\(VJP\) check of .* foo .* failed",
+                         "\(VJP\) check of foo failed",
                          lambda: check_grads(foo, modes=['rev'])(1.0))
 
 def test_check_vjp_2nd_order_fail():
@@ -25,5 +25,5 @@ def bar(x):
     defvjp(bar, lambda ans, vs, gvs, x : lambda g: g * 1.001)
 
     assert_raises_regexp(AssertionError,
-                         "\(VJP\) check of .* vjp_foo .* failed",
+                         "\(VJP\) check of vjp_foo failed",
                          lambda: check_grads(foo, modes=['rev'])(1.0))