tweak poll timeout test

seems extra flaky on 3.11

Author: minrk

zmq/tests/test_poll.py

@@ -103,7 +103,7 @@ def test_no_events(self):
         poller = self.Poller()
         poller.register(s1, zmq.POLLIN | zmq.POLLOUT)
         poller.register(s2, 0)
-        self.assertTrue(s1 in poller)
+        assert s1 in poller
         self.assertFalse(s2 in poller)
         poller.register(s1, 0)
         self.assertFalse(s1 in poller)
@@ -157,25 +157,26 @@ def test_raw(self):
         w.close()
         r.close()
 
+    @mark.flaky(reruns=3)
     def test_timeout(self):
         """make sure Poller.poll timeout has the right units (milliseconds)."""
         s1, s2 = self.create_bound_pair(zmq.PAIR, zmq.PAIR)
         poller = self.Poller()
         poller.register(s1, zmq.POLLIN)
-        tic = time.time()
+        tic = time.perf_counter()
         evt = poller.poll(0.005)
-        toc = time.time()
-        self.assertTrue(toc - tic < 0.1)
-        tic = time.time()
-        evt = poller.poll(5)
-        toc = time.time()
-        self.assertTrue(toc - tic < 0.1)
-        self.assertTrue(toc - tic > 0.001)
-        tic = time.time()
+        toc = time.perf_counter()
+        toc - tic < 0.1
+        tic = time.perf_counter()
+        evt = poller.poll(50)
+        toc = time.perf_counter()
+        assert toc - tic < 0.1
+        assert toc - tic > 0.01
+        tic = time.perf_counter()
         evt = poller.poll(500)
-        toc = time.time()
-        self.assertTrue(toc - tic < 1)
-        self.assertTrue(toc - tic > 0.1)
+        toc = time.perf_counter()
+        assert toc - tic < 1
+        assert toc - tic > 0.1
 
 
 class TestSelect(PollZMQTestCase):
@@ -186,25 +187,25 @@ def test_pair(self):
         wait()
 
         rlist, wlist, xlist = zmq.select([s1, s2], [s1, s2], [s1, s2])
-        self.assertTrue(s1 in wlist)
-        self.assertTrue(s2 in wlist)
-        self.assertTrue(s1 not in rlist)
-        self.assertTrue(s2 not in rlist)
+        assert s1 in wlist
+        assert s2 in wlist
+        assert s1 not in rlist
+        assert s2 not in rlist
 
     @mark.flaky(reruns=3)
     def test_timeout(self):
         """make sure select timeout has the right units (seconds)."""
         s1, s2 = self.create_bound_pair(zmq.PAIR, zmq.PAIR)
-        tic = time.time()
+        tic = time.perf_counter()
         r, w, x = zmq.select([s1, s2], [], [], 0.005)
-        toc = time.time()
-        self.assertTrue(toc - tic < 1)
-        self.assertTrue(toc - tic > 0.001)
-        tic = time.time()
+        toc = time.perf_counter()
+        assert toc - tic < 1
+        assert toc - tic > 0.001
+        tic = time.perf_counter()
         r, w, x = zmq.select([s1, s2], [], [], 0.25)
-        toc = time.time()
-        self.assertTrue(toc - tic < 1)
-        self.assertTrue(toc - tic > 0.1)
+        toc = time.perf_counter()
+        assert toc - tic < 1
+        assert toc - tic > 0.1
 
 
 if have_gevent:
@@ -220,19 +221,19 @@ def test_wakeup(self):
             poller = self.Poller()
             poller.register(s2, zmq.POLLIN)
 
-            tic = time.time()
+            tic = time.perf_counter()
             r = gevent.spawn(lambda: poller.poll(10000))
             s = gevent.spawn(lambda: s1.send(b'msg1'))
             r.join()
-            toc = time.time()
-            self.assertTrue(toc - tic < 1)
+            toc = time.perf_counter()
+            assert toc - tic < 1
 
         def test_socket_poll(self):
             s1, s2 = self.create_bound_pair(zmq.PAIR, zmq.PAIR)
 
-            tic = time.time()
+            tic = time.perf_counter()
             r = gevent.spawn(lambda: s2.poll(10000))
             s = gevent.spawn(lambda: s1.send(b'msg1'))
             r.join()
-            toc = time.time()
-            self.assertTrue(toc - tic < 1)
+            toc = time.perf_counter()
+            assert toc - tic < 1