Move ntp.query retry logic to conftest

conftest.py

@@ -1,6 +1,9 @@
 import sys
 import importlib
+import socket
 
+import pytest
+import jaraco.functools
 from jaraco.context import ExceptionTrap
 
 
@@ -30,3 +33,20 @@ def pywin32_missing():
     ]
     * (sys.version_info > (3, 11))
 )
+
+
+@pytest.fixture(autouse=True)
+def retry_ntp_query(request, monkeypatch):
+    """
+    ntp.query is flaky (by design), so be resilient during tests.
+    """
+    if not request.node.name.endswith('net.ntp.query'):
+        return
+
+    from jaraco.net import ntp
+
+    retry = jaraco.functools.retry(
+        retries=2,
+        trap=socket.timeout,
+    )
+    monkeypatch.setattr(ntp, 'query', retry(ntp.query))

jaraco/net/ntp.py

@@ -13,21 +13,12 @@
 TIME1970 = 0x83AA7E80
 
 
-_flaky = jaraco.functools.retry(
-    retries=2,
-    trap=socket.timeout,
-)
-"""
-ntp.query is flaky (by design), so be resilient to it during tests.
-"""
-
-
 def query(server, force_ipv6=False):
     """
     Return current time from NTP server as a Unix timestamp
 
     >>> import time
-    >>> res = _flaky(query)('us.pool.ntp.org')
+    >>> res = query('pool.ntp.org')
     >>> type(res)
     <class 'int'>
     >>> res > time.time() - 5