From ec069b03a6c41a328869df5d5a4d5a8689185867 Mon Sep 17 00:00:00 2001 From: Matteo Bachetti Date: Tue, 19 Mar 2024 20:22:53 +0100 Subject: [PATCH] Fix enable test --- basie/tests/test_backend.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/basie/tests/test_backend.py b/basie/tests/test_backend.py index c2334a6..7673c3e 100644 --- a/basie/tests/test_backend.py +++ b/basie/tests/test_backend.py @@ -8,7 +8,7 @@ import unittest import copy from io import StringIO - +import numpy as np from basie.backend import * from basie.radiotelescopes import * @@ -30,19 +30,18 @@ def test_roach_backend_bck_file(self): class TestTotalPowerBackend(unittest.TestCase): def setUp(self): - self.backend = TotalPowerBackend("TP", 10, 10, 300) + self.backend = TotalPowerBackend("TP", 10, 10, 300, feeds="1,2") def test_total_power_set_sections(self): n_sections = 2 self.backend.set_sections(n_sections) self.assertEqual(len(self.backend.sections), n_sections) - @pytest.mark.xfail + def test_total_power_set_sections_enable(self): self.backend.set_sections(2) instructions = StringIO(str(self.backend._get_backend_instructions())) lines = instructions.readlines() - enable_line = lines[-1].strip() - self.assertTrue(enable_line.startswith("enable")) + self.assertTrue(np.any(["enable=1,2" in line for line in lines])) def test_set_addition(self): # https://github.com/discos/basie/issues/28