From 0da320eff22bb2e679dfc3582d34fe508c8fa45e Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Sat, 8 Jun 2024 16:41:03 +0800 Subject: [PATCH] add delete_section --- tests/test_fits_header_ops.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/tests/test_fits_header_ops.py b/tests/test_fits_header_ops.py index 3f0d641..efc8b25 100644 --- a/tests/test_fits_header_ops.py +++ b/tests/test_fits_header_ops.py @@ -8,11 +8,12 @@ Modified-History: 2023-12-15, Bo Zhang, add TestFitsHeaderOps.test_append_header 2023-12-15, Bo Zhang, add TestFitsHeaderOps.test_reformat_header """ + import unittest from astropy.io import fits -from csst_common.io import append_header, reformat_header +from csst_common.io import append_header, reformat_header, delete_section class TestFitsHeaderOps(unittest.TestCase): @@ -100,3 +101,20 @@ class TestFitsHeaderOps(unittest.TestCase): tuple(h_rfmt.keys()), ("COMMENT", "COMMENT", "COMMENT", "A", "SIMPLE", "NAXIS1"), ) + + def test_delete_section(self): + h = fits.Header() + h.add_comment("A") + h.add_comment("B") + h.add_comment("C") + h.set("SIMPLE", True) + h.set("NAXIS1", 1) + h.add_comment("======") + h.add_comment("WCS") + h.add_comment("======") + h.set("A", 1) + h.add_comment("======") + h.add_comment("======") + + h_del = delete_section(h, title="WCS") + self.assertEqual(len(h_del.cards), 7) -- GitLab