summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--numpy/core/tests/test_umath.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/numpy/core/tests/test_umath.py b/numpy/core/tests/test_umath.py
index 6c6462584..c82e5af7c 100644
--- a/numpy/core/tests/test_umath.py
+++ b/numpy/core/tests/test_umath.py
@@ -234,6 +234,11 @@ def test_arctan2_special_values():
def assert_arctan2_isnzero(x, y):
assert ncu.arctan2(x, y) == 0 and np.signbit(ncu.arctan2(x, y))
+ # atan2(1, 1) returns pi/4.
+ yield assert_almost_equal, ncu.arctan2(1, 1), 0.25 * np.pi
+ yield assert_almost_equal, ncu.arctan2(-1, 1), -0.25 * np.pi
+ yield assert_almost_equal, ncu.arctan2(1, -1), 0.75 * np.pi
+
# atan2(+-0, -0) returns +-pi.
yield assert_almost_equal, ncu.arctan2(np.PZERO, np.NZERO), np.pi
yield assert_almost_equal, ncu.arctan2(np.NZERO, np.NZERO), -np.pi