Update Lib/test/test_math.py

Co-authored-by: Sergey B Kirpichev <[email protected]>
iritkatriel · Oct 18, 2024 · a22534e · a22534e
1 parent 1340811
commit a22534e
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/Lib/test/test_math.py b/Lib/test/test_math.py
@@ -1902,19 +1902,11 @@ def testTan(self):
         self.assertTrue(math.isnan(math.tan(NAN)))
 
         try:
-            tan = math.tan(INF)
+            self.assertTrue(math.isnan(math.tan(INF)))
+            self.assertTrue(math.isnan(math.tan(NINF)))
         except ValueError:
-            pass
-        else:
-            self.assertTrue(math.isnan(tan))
-
-        try:
-            ntan = math.tan(NINF)
-        except ValueError:
-            pass
-        else:
-            self.assertTrue(math.isnan(ntan))
-
+            self.assertRaises(ValueError, math.tan, INF)
+            self.assertRaises(ValueError, math.tan, NINF)
     def testTanh(self):
         self.assertRaises(TypeError, math.tanh)
         self.ftest('tanh(0)', math.tanh(0), 0)