From 120db2dd875f001a2d79629c145f4b11a5a75434 Mon Sep 17 00:00:00 2001
From: Jan David Mol <mol@astron.nl>
Date: Wed, 23 Oct 2024 13:30:35 +0200
Subject: [PATCH] support more than 488 beamlets in tests
---
.../test/devices/sdp/test_beamlet_device.py | 28 ++++++++++++++++---
1 file changed, 24 insertions(+), 4 deletions(-)
diff --git a/tangostationcontrol/test/devices/sdp/test_beamlet_device.py b/tangostationcontrol/test/devices/sdp/test_beamlet_device.py
index 0e8ef1a10..f5166079e 100644
--- a/tangostationcontrol/test/devices/sdp/test_beamlet_device.py
+++ b/tangostationcontrol/test/devices/sdp/test_beamlet_device.py
@@ -177,7 +177,12 @@ class TestBeamletDevice(base.TestCase):
def test_beamlet_frequencies_basic(self):
beamlet_subbands = numpy.array(
- [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
+ [
+ [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol]
+ * A_pn
+ ]
+ * N_pn,
+ dtype=numpy.uint32,
)
subband_indices = numpy.array(
[[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
@@ -190,7 +195,12 @@ class TestBeamletDevice(base.TestCase):
"""Test different settings for the FPGAs to verify the ordering."""
beamlet_subbands = numpy.array(
- [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
+ [
+ [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol]
+ * A_pn
+ ]
+ * N_pn,
+ dtype=numpy.uint32,
)
subband_indices = numpy.array(
[[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
@@ -206,7 +216,12 @@ class TestBeamletDevice(base.TestCase):
"""Test different settings for the inputs of the FPGAs to verify the ordering."""
beamlet_subbands = numpy.array(
- [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
+ [
+ [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol]
+ * A_pn
+ ]
+ * N_pn,
+ dtype=numpy.uint32,
)
subband_indices = numpy.array(
[[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
@@ -222,7 +237,12 @@ class TestBeamletDevice(base.TestCase):
"""Test a shuffled order of subbands."""
beamlet_subbands = numpy.array(
- [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
+ [
+ [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol]
+ * A_pn
+ ]
+ * N_pn,
+ dtype=numpy.uint32,
)
subband_indices = numpy.array(
[[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32
--
GitLab