diff --git a/tests/test_utils_pvpreload.py b/tests/test_utils_pvpreload.py index 7292123df..8d4bd3042 100644 --- a/tests/test_utils_pvpreload.py +++ b/tests/test_utils_pvpreload.py @@ -11,6 +11,8 @@ from freezegun import freeze_time from datetime import datetime, timedelta from slic.utils.hastyepics import get_pv from epics.pv import _PVcache_ +from slic.utils.pv import PV + import sys import os @@ -172,11 +174,8 @@ def test_offload(tmp_path, caplog): pvname_4 = "TEST:PV4" # Create 2 PVs - pv_3 = epics.PV(pvname_3, connection_timeout=2) - pv_4 = epics.PV(pvname_4, connection_timeout=2) - - pv_3.wait_for_connection(timeout=2) - pv_4.wait_for_connection(timeout=2) + pv_3 = PV(pvname_3, connection_timeout=2) + pv_4 = PV(pvname_4, connection_timeout=2) assert is_pv_in_cache(pvname_3) assert is_pv_in_cache(pvname_4)