better error message for adding sphere basis with different raddi (#224)

DedalusProject · Feb 6, 2023 · 5b52e29 · 5b52e29
1 parent 2648fcc
commit 5b52e29
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dedalus/core/basis.py b/dedalus/core/basis.py
@@ -2761,6 +2761,8 @@ def __add__(self, other):
             if self.radius == other.radius:
                 shape = tuple(np.maximum(self.shape, other.shape))
                 return SphereBasis(self.coordsystem, shape, radius=self.radius, dealias=self.dealias, dtype=self.dtype)
+            else:
+                raise NotImplementedError(f"Cannot add two sphere bases with different radii {self.radius} and {other.radius}.")
         return NotImplemented
 
     def __mul__(self, other):
@@ -2772,6 +2774,8 @@ def __mul__(self, other):
             if self.radius == other.radius:
                 shape = tuple(np.maximum(self.shape, other.shape))
                 return SphereBasis(self.coordsystem, shape, radius=self.radius, dealias=self.dealias, dtype=self.dtype)
+            else:
+                raise NotImplementedError(f"Cannot multiply two sphere bases with different radii {self.radius} and {other.radius}.")
         return NotImplemented
 
     def __matmul__(self, other):