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

add get_proc_info

parent ad677026

csst_common/io.py

+1 −1
Original line number Diff line number Diff line
@@ -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)

tests/test_fits_header_ops.py

+5 −1
Original line number Diff line number Diff line
@@ -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"], "-")