Fixed stupid mistake

zwicker-group · Aug 16, 2024 · e6ddf00 · e6ddf00
1 parent 5ea8966
commit e6ddf00
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/pdes/test_pde_class.py b/tests/pdes/test_pde_class.py
@@ -288,7 +288,9 @@ def test_pde_consts():
 
     eq = PDE({"a": "laplace(b)"}, consts={"b": 3})
     with pytest.raises(
-        AttributeError if os.environ["NUMBA_DISABLE_JIT"] == "1" else nb.TypingError
+        AttributeError
+        if os.environ.get("NUMBA_DISABLE_JIT", "0") == "1"
+        else nb.TypingError
     ):
         eq.evolution_rate(field)