From d396d847864a76b45cdcb268065d76b406295833 Mon Sep 17 00:00:00 2001 From: Martin Bauer <martin.bauer@fau.de> Date: Wed, 26 Jul 2017 15:03:36 +0200 Subject: [PATCH] sympy 1.1 compatibility --- boundaries/boundaryconditions.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/boundaries/boundaryconditions.py b/boundaries/boundaryconditions.py index 2849747e..efc892b6 100644 --- a/boundaries/boundaryconditions.py +++ b/boundaries/boundaryconditions.py @@ -1,6 +1,7 @@ import sympy as sp from lbmpy.simplificationfactory import createSimplificationStrategy +from pystencils.astnodes import SympyAssignment from pystencils.sympyextensions import getSymmetricPart from pystencils import Field from lbmpy.boundaries.boundary_kernel import offsetFromDir, weightOfDirection, invDir @@ -159,7 +160,9 @@ class FixedDensity(Boundary): subExprs = [sp.Eq(eq.lhs, transformedDensity if eq.lhs == densitySymbol else eq.rhs) for eq in symmetricEq.subexpressions] - return subExprs + [sp.Eq(pdfField[neighbor](inverseDir), 2 * eq_component - pdfField(directionSymbol))] + + return subExprs + [SympyAssignment(pdfField[neighbor](inverseDir), + 2 * eq_component - pdfField(directionSymbol))] class NeumannByCopy(Boundary): @@ -173,3 +176,10 @@ class NeumannByCopy(Boundary): def __eq__(self, other): return type(other) == NeumannByCopy + + +if __name__ == '__main__': + from lbmpy.scenarios import * + sc = createForceDrivenChannel(domainSize=[60, 30]) + sc.boundaryHandling.setBoundary(FixedDensity(1.01), makeSlice[0.2:0.25, :]) + sc.boundaryHandling.prepare() -- GitLab