From e6ddf00e7cd27112fd8b553b118f1bffc29f3061 Mon Sep 17 00:00:00 2001
From: David Zwicker <david.zwicker@ds.mpg.de>
Date: Fri, 16 Aug 2024 17:13:10 +0200
Subject: [PATCH] Fixed stupid mistake

---
 tests/pdes/test_pde_class.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/pdes/test_pde_class.py b/tests/pdes/test_pde_class.py
index d26ae7a0..4796d26a 100644
--- 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)