Fix test

ColCarroll · Jul 7, 2019 · 811a652 · 811a652
1 parent e829647
commit 811a652
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arviz/tests/test_utils.py b/arviz/tests/test_utils.py
@@ -123,7 +123,7 @@ def test_conditional_jit_numba_decorator_keyword(monkeypatch):
 
     def jit(**kwargs):
         """overwrite numba.jit function"""
-        return lambda x: (x(), kwargs)
+        return lambda fn: lambda: (fn(), kwargs)
 
     numba_mock.jit = jit
 
@@ -133,7 +133,7 @@ def placeholder_func():
         return "output"
 
     # pylint: disable=unpacking-non-sequence
-    function_results, wrapper_result = placeholder_func
+    function_results, wrapper_result = placeholder_func()
     assert wrapper_result == {"keyword_argument": "A keyword argument"}
     assert function_results == "output"