diff --git a/src/one-body-sample.py b/src/one-body-sample.py index 00bf779977c63482b0dbaa57e40d497dd6125960..313edd816bca47204f1522934542ca5a3ed1a69a 100644 --- a/src/one-body-sample.py +++ b/src/one-body-sample.py @@ -1,15 +1,15 @@ class neumannCondition: def __call__(self, x): return 0 - # fst = 3e3 - # snd = 5e3 - # trd = 3e3 - # if x < 1.0/3: - # return fst * x - # elif x < 2.0/3: - # return snd * (x - 1.0/3) + fst * 1.0/3 - # else: - # return trd * (x - 2.0/3) + (fst + snd) * 1.0/3 + fst = 3e3 + snd = 5e3 + trd = 3e3 + if x < 1.0/3: + return fst * x + elif x < 2.0/3: + return snd * (x - 1.0/3) + fst * 1.0/3 + else: + return trd * (x - 2.0/3) + (fst + snd) * 1.0/3 class dirichletCondition: def __call__(self, x):