Skip to content

Commit ce380e9

Browse files
committed
fix test for new pymbolic
1 parent 714da54 commit ce380e9

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/test_pde_system_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -429,7 +429,7 @@ def test_restoring_target_attributes():
429429

430430

431431
def test_int_gs_in_densities():
432-
from pymbolic.primitives import Variable
432+
from pymbolic.primitives import Variable, Product
433433
dim = 3
434434
laplace_knl = LaplaceKernel(dim)
435435
density = Variable("density")
@@ -449,9 +449,10 @@ def test_int_gs_in_densities():
449449
result = merge_int_g_exprs([int_g1])
450450

451451
source_kernels = [AxisSourceDerivative(0, laplace_knl), laplace_knl]
452+
452453
densities = [
453-
(-1)*int_g_vec(AxisTargetDerivative(1, laplace_knl),
454-
density, qbx_forced_limit=1) * (-2),
454+
Product((-1, Product((int_g_vec(AxisTargetDerivative(1, laplace_knl),
455+
density, qbx_forced_limit=1), -2)))),
455456
int_g_vec(AxisTargetDerivative(2, laplace_knl),
456457
density, qbx_forced_limit=1) * (-1)
457458
]
@@ -460,7 +461,6 @@ def test_int_gs_in_densities():
460461
densities=tuple(densities),
461462
qbx_forced_limit=1)
462463

463-
print(result[0])
464464
assert result[0] == int_g3
465465

466466

0 commit comments

Comments
 (0)