diff --git a/sumpy/toys.py b/sumpy/toys.py index 64431538e9bcd2989492ff2a2e2c66bcaa753ea6..512dc3a1486883517f94cda0e08410cd453a7b27 100644 --- a/sumpy/toys.py +++ b/sumpy/toys.py @@ -272,7 +272,7 @@ class OneOnBallPotential(PotentialSource): def eval(self, targets): dist_vec = targets - self.center[:, np.newaxis] - return (np.sum(dist_vec**2, axis=0) < self.radius).astype(np.float64) + return (np.sum(dist_vec**2, axis=0) < self.radius**2).astype(np.float64) class PointSources(PotentialSource):