diff --git a/tangostationcontrol/tangostationcontrol/integration_test/recv_cluster/test_recv_cluster.py b/tangostationcontrol/tangostationcontrol/integration_test/recv_cluster/test_recv_cluster.py index 4db2ac730e5eb9f18cce477d071cdd13abccd2d4..292ced63af786fe8c1689d404d6ddc010a7b5a97 100644 --- a/tangostationcontrol/tangostationcontrol/integration_test/recv_cluster/test_recv_cluster.py +++ b/tangostationcontrol/tangostationcontrol/integration_test/recv_cluster/test_recv_cluster.py @@ -16,13 +16,15 @@ from tango import DevState from tangostationcontrol.integration_test import base from tangostationcontrol.integration_test.device_proxy import TestDeviceProxy +from tangostationcontrol.devices.antennafield import AntennaQuality, AntennaUse logger = logging.getLogger() class TestRecvCluster(base.IntegrationTestCase): - POINTING_DIRECTION = numpy.array([["J2000", "0deg", "0deg"]] * 96).flatten() + NR_TILES = 48 + POINTING_DIRECTION = numpy.array([["J2000","0deg","0deg"]] * NR_TILES).flatten() def setUp(self): @@ -48,10 +50,17 @@ class TestRecvCluster(base.IntegrationTestCase): self.assertTrue(proxy.state() is DevState.ON) for proxy in antenna_field_proxies: - proxy.off() - self.assertTrue(proxy.state() is DevState.OFF) - proxy.warm_boot() - proxy.set_defaults() + # setup AntennaField + NR_TILES = 48 + antennafield_proxy = TestDeviceProxy(f"STAT/AntennaField/{i}") + control_mapping = [[1,i] for i in range(NR_TILES)] + antenna_qualities = numpy.array([AntennaQuality.OK] * 96) + antenna_use = numpy.array([AntennaUse.AUTO] * 96) + antennafield_proxy.put_property({"RECV_devices": [f"STAT/RECV/{i}"], + "Control_to_RECV_mapping": numpy.array(control_mapping).flatten(), + 'Antenna_Quality': antenna_qualities, 'Antenna_Use': antenna_use}) + antennafield_proxy.off() + antennafield_proxy.boot() self.assertTrue(proxy.state() is DevState.ON) for proxy in beam_proxies: @@ -59,6 +68,7 @@ class TestRecvCluster(base.IntegrationTestCase): self.assertTrue(proxy.state() is DevState.OFF) proxy.warm_boot() proxy.set_defaults() + proxy.Tracking_enabled_RW = False self.assertTrue(proxy.state() is DevState.ON) results = []