diff --git a/tangostationcontrol/tangostationcontrol/test/SDP_XST_statistics_packets_multiple_subbands.bin b/tangostationcontrol/tangostationcontrol/test/SDP_XST_statistics_packets_multiple_subbands.bin new file mode 100644 index 0000000000000000000000000000000000000000..4fd980fd13fc166c29213006d2935b4eb30d3435 Binary files /dev/null and b/tangostationcontrol/tangostationcontrol/test/SDP_XST_statistics_packets_multiple_subbands.bin differ diff --git a/tangostationcontrol/tangostationcontrol/test/test_statistics_writer.py b/tangostationcontrol/tangostationcontrol/test/test_statistics_writer.py index 4b10230c8fca8c0773d085c6e42719b74dd7b2c3..8ce24f7bc4a7d10da08801348c8cdc6bf3af49f1 100644 --- a/tangostationcontrol/tangostationcontrol/test/test_statistics_writer.py +++ b/tangostationcontrol/tangostationcontrol/test/test_statistics_writer.py @@ -10,7 +10,7 @@ from tangostationcontrol.test import base from tangostationcontrol.statistics_writer import statistics_writer import sys -from os.path import dirname +from os.path import dirname, isfile from tempfile import TemporaryDirectory from unittest import mock @@ -22,9 +22,27 @@ class TestStatisticsWriter(base.TestCase): with self.assertRaises(SystemExit): statistics_writer.main() + # check if file was written + self.assertTrue(isfile(f"{tmpdir}/SST_2021-09-20-12-17-40.h5")) + def test_xst(self): with TemporaryDirectory() as tmpdir: new_sys_argv = [sys.argv[0], "--mode", "XST", "--file", dirname(__file__) + "/SDP_XST_statistics_packets.bin", "--output_dir", tmpdir] with mock.patch.object(statistics_writer.sys, 'argv', new_sys_argv): with self.assertRaises(SystemExit): statistics_writer.main() + + # check if file was written + self.assertTrue(isfile(f"{tmpdir}/XST_SB102_2021-09-13-13-21-32.h5")) + + def test_xst_multiple_subbands(self): + with TemporaryDirectory() as tmpdir: + new_sys_argv = [sys.argv[0], "--mode", "XST", "--file", dirname(__file__) + "/SDP_XST_statistics_packets_multiple_subbands.bin", "--output_dir", tmpdir] + with mock.patch.object(statistics_writer.sys, 'argv', new_sys_argv): + with self.assertRaises(SystemExit): + statistics_writer.main() + + # check if files were written + self.assertTrue(isfile(f"{tmpdir}/XST_SB102_2021-09-13-13-21-32.h5")) + self.assertTrue(isfile(f"{tmpdir}/XST_SB103_2021-09-13-13-21-32.h5")) +