Add the copysign ufunc

Author: juntyr

quaddtype/numpy_quaddtype/src/ops.hpp

@@ -339,6 +339,12 @@ quad_atan2(Sleef_quad *in1, Sleef_quad *in2)
     return Sleef_atan2q1_u10(*in1, *in2);
 }
 
+static inline Sleef_quad
+quad_copysign(Sleef_quad *in1, Sleef_quad *in2)
+{
+    return Sleef_copysignq1(*in1, *in2);
+}
+
 // Binary long double operations
 typedef long double (*binary_op_longdouble_def)(long double *, long double *);
 
@@ -396,6 +402,12 @@ ld_atan2(long double *in1, long double *in2)
     return atan2l(*in1, *in2);
 }
 
+static inline long double
+ld_copysign(long double *in1, long double *in2)
+{
+    return copysignl(*in1, *in2);
+}
+
 // comparison quad functions
 typedef npy_bool (*cmp_quad_def)(const Sleef_quad *, const Sleef_quad *);
 

quaddtype/numpy_quaddtype/src/umath.cpp

@@ -556,6 +556,9 @@ init_quad_binary_ops(PyObject *numpy)
     if (create_quad_binary_ufunc<quad_atan2, ld_atan2>(numpy, "arctan2") < 0) {
         return -1;
     }
+    if (create_quad_binary_ufunc<quad_copysign, ld_copysign>(numpy, "copysign") < 0) {
+        return -1;
+    }
     return 0;
 }
 

quaddtype/tests/test_quaddtype.py

@@ -17,13 +17,13 @@ def test_basic_equality():
         "12.0") == QuadPrecision("12.00")
 
 
-@pytest.mark.parametrize("op", ["add", "sub", "mul", "truediv", "pow"])
+@pytest.mark.parametrize("op", ["add", "sub", "mul", "truediv", "pow", "copysign"])
 @pytest.mark.parametrize("other", ["3.0", "12.5", "100.0", "0.0", "-0.0", "inf", "-inf", "nan", "-nan"])
 def test_binary_ops(op, other):
     if op == "truediv" and float(other) == 0:
         pytest.xfail("float division by zero")
 
-    op_func = getattr(operator, op)
+    op_func = getattr(operator, op, None) or getattr(np, op)
     quad_a = QuadPrecision("12.5")
     quad_b = QuadPrecision(other)
     float_a = 12.5