test also div terminal expr

pyccel · Feb 26, 2025 · 22c89c9 · 22c89c9
1 parent 4aeb886
commit 22c89c9
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/sympde/expr/tests/test_expr.py b/sympde/expr/tests/test_expr.py
@@ -1904,14 +1904,17 @@ def test_terminal_expressions_for_navier_stokes():
     pe = sin(pi*y)
     # ...
 
-    # Verify that div(u) = 0
-    assert (ux.diff(x) + uy.diff(y)).simplify() == 0
-
     # ... Compute right-hand side
     a = TerminalExpr(-mu*laplace(ue), domain)
     b = TerminalExpr(    grad(ue), domain)
     c = TerminalExpr(    grad(pe), domain)
-
+
+    # Verify that div(u) = 0
+    assert (ux.diff(x) + uy.diff(y)).simplify() == 0
+
+    d = TerminalExpr(div(ue), domain)
+    assert d.simplify() == 0
+
     f = (a + b.T*ue + c).simplify()
 
     fx = -mu*(ux.diff(x, 2) + ux.diff(y, 2)) + ux*ux.diff(x) + uy*ux.diff(y) + pe.diff(x)