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(
def get_proc_info(**kwargs):
cards = [("BUILD", os.getenv("BUILD"), "")]
cards = [("BUILD", os.getenv("BUILD", default="-"), "")]
for k, v in kwargs.items():
cards.append((k, v, ""))
header = fits.Header(cards=cards)
......
......@@ -13,7 +13,7 @@ import unittest
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):
......@@ -118,3 +118,7 @@ class TestFitsHeaderOps(unittest.TestCase):
h_del = delete_section(h, title="WCS")
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