sorting functions

Author: lucascolley

src/pint_array/__init__.py

@@ -556,6 +556,22 @@ def matrix_transpose(x):
 
     mod.matrix_transpose = matrix_transpose
 
+    ## Sorting Functions ##
+    def get_sort_fun(func_str):
+        def sort_fun(x, /, **kwargs):
+            x = asarray(x)
+            magnitude = xp.asarray(x.magnitude, copy=True)
+            xp_func = getattr(xp, func_str)
+            magnitude = xp_func(magnitude, **kwargs)
+            units = x.units if func_str == "sort" else None
+            return ArrayUnitQuantity(magnitude, units)
+
+        return sort_fun
+
+    sort_names = ["sort", "argsort"]
+    for name in sort_names:
+        setattr(mod, name, get_sort_fun(name))
+
     ## Set Functions ##
     def get_set_fun(func_str):
         def set_fun(x, /):