Fixed bug where units weren't being checked properly for `named_array…

…s.random` functions. (#76)
sun-data · Oct 3, 2024 · d351498 · d351498
1 parent 3878330
commit d351498
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/named_arrays/_scalars/scalar_named_array_functions.py b/named_arrays/_scalars/scalar_named_array_functions.py
@@ -382,12 +382,12 @@ def random(
 
     if unit is not None:
         args = tuple(
-            arg.value if isinstance(arg, u.Quantity)
+            arg.to_value(unit) if isinstance(arg, u.Quantity)
             else (arg << u.dimensionless_unscaled).to_value(unit)
             for arg in args
         )
         kwargs = {
-            k: kwargs[k].value if isinstance(kwargs[k], u.Quantity)
+            k: kwargs[k].to_value(unit) if isinstance(kwargs[k], u.Quantity)
             else (kwargs[k] << u.dimensionless_unscaled).to_value(unit)
             for k in kwargs
         }