fix expected np.arctan2() exception for numpy 1.21

mgorny · hugovk · commit 4809140af034 · 2021-09-25T22:44:42.000+03:00
NumPy 1.21 has changed the exception type for incorrect np.arctan2() arguments from ValueError to TypeError. Adjust the test appropriately. Fixes python-quantities#190
diff --git a/quantities/tests/test_umath.py b/quantities/tests/test_umath.py
@@ -226,7 +226,9 @@ def test_arctan2(self):
             np.arctan2(3*pq.V, 3*pq.V),
             np.radians(45)*pq.dimensionless
             )
-        self.assertRaises(ValueError, np.arctan2, (1*pq.m, 1*pq.m))
+        # NumPy <1.21 raises ValueError
+        # NumPy >=1.21 raises TypeError
+        self.assertRaises((TypeError, ValueError), np.arctan2, (1*pq.m, 1*pq.m))
 
     def test_hypot(self):
         self.assertQuantityEqual(np.hypot(3 * pq.m, 4 * pq.m),  5 * pq.m)
