fix test

chaobrain · Dec 28, 2024 · 50964eb · 50964eb
1 parent 31f1946
commit 50964eb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/brainunit/_base.py b/brainunit/_base.py
@@ -4922,6 +4922,13 @@ def _remove_unit(fname, n, unit, v):
                 f"object for argument '{n}' but got '{v}'"
             )
 
+    elif unit == 1:
+        if isinstance(v, Quantity):
+            raise TypeError(
+                f"Function '{fname}' expected a Number object for argument '{n}' but got '{v}'"
+            )
+        return v
+
     else:
         raise TypeError(
             f"Function '{fname}' expected a target unit object or"