Commit 86cff58d authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add get_proc_info

parent ad677026
Pipeline #9242 passed with stage
...@@ -195,7 +195,7 @@ def delete_section( ...@@ -195,7 +195,7 @@ def delete_section(
def get_proc_info(**kwargs): def get_proc_info(**kwargs):
cards = [("BUILD", os.getenv("BUILD"), "")] cards = [("BUILD", os.getenv("BUILD", default="-"), "")]
for k, v in kwargs.items(): for k, v in kwargs.items():
cards.append((k, v, "")) cards.append((k, v, ""))
header = fits.Header(cards=cards) header = fits.Header(cards=cards)
......
...@@ -13,7 +13,7 @@ import unittest ...@@ -13,7 +13,7 @@ import unittest
from astropy.io import fits from astropy.io import fits
from csst_common.io import append_header, reformat_header, delete_section from csst_common.io import append_header, reformat_header, delete_section, get_proc_info
class TestFitsHeaderOps(unittest.TestCase): class TestFitsHeaderOps(unittest.TestCase):
...@@ -118,3 +118,7 @@ class TestFitsHeaderOps(unittest.TestCase): ...@@ -118,3 +118,7 @@ class TestFitsHeaderOps(unittest.TestCase):
h_del = delete_section(h, title="WCS") h_del = delete_section(h, title="WCS")
self.assertEqual(len(h_del.cards), 7) self.assertEqual(len(h_del.cards), 7)
def test_get_proc_info(self):
proc_info = get_proc_info()
self.assertEquals(proc_info["BUILD"], "-")
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