diff --git a/tangostationcontrol/tangostationcontrol/test/devices/test_beam_device.py b/tangostationcontrol/tangostationcontrol/test/devices/test_beam_device.py index 07ac6ebdf629664bf908ca7d8df75262906c4d4f..31e2d1c362d27e3843b77216ce3797b3b51da2b8 100644 --- a/tangostationcontrol/tangostationcontrol/test/devices/test_beam_device.py +++ b/tangostationcontrol/tangostationcontrol/test/devices/test_beam_device.py @@ -9,8 +9,9 @@ from tango import DevState from tango.test_context import DeviceTestContext +import tango -from tangostationcontrol.devices import beam +from tangostationcontrol.devices import beam, lofar_device import numpy import mock @@ -25,10 +26,11 @@ class TestBeamDevice(base.TestCase): # Patch DeviceProxy to allow making the proxies during initialisation # that we otherwise avoid using - proxy_patcher = mock.patch.object( - beam, 'DeviceProxy') - self.m_init = proxy_patcher.start() - self.addCleanup(proxy_patcher.stop) + for device in [beam, lofar_device]: + proxy_patcher = mock.patch.object( + device, 'DeviceProxy') + proxy_patcher.start() + self.addCleanup(proxy_patcher.stop) def test_get_pointing_directions(self): """Verify can read pointings attribute and length matches without err"""