Support multiple results

brainunit/_base.py

@@ -4492,6 +4492,12 @@ def new_f(*args, **kwds):
                     newkeyset[n] = v
 
             result = f(**newkeyset)
+            if isinstance(result, tuple):
+                assert isinstance(au["result"], tuple), "The return value of the function is a tuple, but the decorator expected a single unit."
+                result = tuple(
+                    Quantity(r, unit=au["result"][i]) if isinstance(au["result"][i], Unit) else r
+                    for i, r in enumerate(result)
+                )
             if "result" in au:
                 specific_unit = au["result"]
                 if specific_unit == bool:

brainunit/_base_test.py

@@ -1367,6 +1367,16 @@ def c_function(a, b):
         with pytest.raises(TypeError):
             c_function(1 * mV, 1)
 
+        # Multiple results
+        @u.handle_units(result=(second, volt))
+        def d_function():
+            return 5, 3
+
+        # Should work (returns second)
+        assert d_function()[0] == 5 * second
+        assert d_function()[1] == 3 * volt
+
+
 
 def test_str_repr():
     """