Commit 396f14f7 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add append_meta

parent 08f9b3f8
Pipeline #10166 failed with stage
in 0 seconds
...@@ -19,6 +19,7 @@ from csst_common.io import ( ...@@ -19,6 +19,7 @@ from csst_common.io import (
delete_section, delete_section,
get_proc_info, get_proc_info,
generate_meta, generate_meta,
append_meta,
) )
...@@ -136,3 +137,10 @@ class TestFitsHeaderOps(unittest.TestCase): ...@@ -136,3 +137,10 @@ class TestFitsHeaderOps(unittest.TestCase):
def test_generate_meta_with_invalid_args(self): def test_generate_meta_with_invalid_args(self):
with self.assertRaises(AssertionError): with self.assertRaises(AssertionError):
generate_meta(obs_id=1) generate_meta(obs_id=1)
def test_append_meta(self):
hdulist = fits.HDUList()
hdulist.append(fits.PrimaryHDU())
hdulist = append_meta(hdulist, obs_id="123456")
self.assertIn("OBS_ID", hdulist[0].header)
self.assertEqual(hdulist[0].header["OBS_ID"], "123456")
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment