diff --git a/finitedifferences.py b/finitedifferences.py index 171260cdc1b7dc1852dd02e8dff7db4dadfa22aa..423b62064395fb7797d661dcf6461cbb214722e4 100644 --- a/finitedifferences.py +++ b/finitedifferences.py @@ -115,14 +115,14 @@ def discretizeDivergence(vectorTerm, symbolsToFieldDict, dx): >>> gradX = grad(x, dim=3) >>> f = Field.createGeneric('f', spatialDimensions=3) >>> sp.simplify(discretizeDivergence(gradX, {x : f}, dx)) - (f_B - 6*f_C + f_E + f_N + f_S + f_T + f_W)/dx + (f_B - 6*f_C + f_E + f_N + f_S + f_T + f_W)/dx**2 """ dim = len(vectorTerm) result = 0 for d in range(dim): for offset in [-1, 1]: result += offset * discretizeStaggered(vectorTerm[d], symbolsToFieldDict, d, offset, dx, dim) - return result + return result / dx def __upDownOffsets(d, dim):