Commit 11b366fd authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 788497ee

csst_common/io.py

+4 −3
Original line number Original line Diff line number Diff line
@@ -9,6 +9,7 @@ Modified-History: 
    2023-12-15, Bo Zhang, added verify_checksum and append_header, add module header

    2023-12-15, Bo Zhang, added verify_checksum and append_header, add module header

    2023-12-16, Bo Zhang, fixed a bug in finding the first key when reformatting

    2023-12-16, Bo Zhang, fixed a bug in finding the first key when reformatting

    2024-06-08, Bo Zhang, added delete_section

    2024-06-08, Bo Zhang, added delete_section

    2025-08-05, Bo Zhang, added generate_meta

"""

"""





import os

import os
@@ -16,14 +17,14 @@ import json 
import bisect

import bisect

import warnings

import warnings

from copy import deepcopy

from copy import deepcopy

from typing import Optional

from typing import Optional, Any





from astropy.io import fits

from astropy.io import fits









REQUIRED_KEYS_IN_META = dict(

REQUIRED_KEYS_IN_META = dict(

    # 观测数据相关的ID参数

    # 观测数据相关的ID参数

    dataset="__instrument__",

    dataset="__dataset__",

    instrument="__instrument__",

    instrument="__instrument__",

    obs_type="__obs_type__",

    obs_type="__obs_type__",

    obs_group="__obs_group__",

    obs_group="__obs_group__",
@@ -56,7 +57,7 @@ def generate_meta(**kwargs) -> fits.Header: 




    Parameters

    Parameters

    ----------

    ----------

    kwargs : dict

    kwargs : Any

        Meta data.

        Meta data.





    Returns

    Returns

tests/test_fits_header_ops.py

+11 −1
Original line number Original line Diff line number Diff line
@@ -13,7 +13,13 @@ import unittest 




from astropy.io import fits

from astropy.io import fits





from csst_common.io import append_header, reformat_header, delete_section, get_proc_info

from csst_common.io import (

    append_header,

    reformat_header,

    delete_section,

    get_proc_info,

    generate_meta,

)









class TestFitsHeaderOps(unittest.TestCase):

class TestFitsHeaderOps(unittest.TestCase):
@@ -122,3 +128,7 @@ class TestFitsHeaderOps(unittest.TestCase): 
    def test_get_proc_info(self):

    def test_get_proc_info(self):

        proc_info = get_proc_info()

        proc_info = get_proc_info()

        self.assertEquals(proc_info["BUILD"], "-")

        self.assertEquals(proc_info["BUILD"], "-")



    def test_generate_meta(self):

        h = generate_meta()

        self.assertIsInstance(h, fits.Header)