Skip to content
test_mbi_level1.py 1.34 KiB
Newer Older
Wei Shoulin's avatar
c3
Wei Shoulin committed
import os
import unittest
from astropy.io import fits

Wei Shoulin's avatar
Wei Shoulin committed
from csst_dfs_api.mbi.level1 import Level1DataApi
Wei Shoulin's avatar
c3
Wei Shoulin committed

Wei Shoulin's avatar
Wei Shoulin committed
class MBILevel1DataTestCase(unittest.TestCase):
Wei Shoulin's avatar
c3
Wei Shoulin committed

    def setUp(self):
        self.api = Level1DataApi()

    def test_find(self):
Wei Shoulin's avatar
Wei Shoulin committed
        recs = self.api.find(
            level0_id='000001201',
Wei Shoulin's avatar
c3
Wei Shoulin committed
            create_time = ("2021-06-01 11:12:13","2021-06-08 11:12:13"))
        print('find:', recs)

    def test_get(self):
        rec = self.api.get(id = 2)
        print('get:', rec)

    def test_update_proc_status(self):
        rec = self.api.update_proc_status(id = 2, status = 4)
        print('update_proc_status:', rec)

    def test_update_qc1_status(self):
        rec = self.api.update_qc1_status(id = 2, status = 7)
        print('update_qc1_status:', rec)

    def test_write(self):
Wei Shoulin's avatar
Wei Shoulin committed
        rec = self.api.write(
            level0_id='000001201', 
Wei Shoulin's avatar
c3
Wei Shoulin committed
            data_type = "sci",
            cor_sci_id = 1,
            prc_params = "/opt/dddasd.params",
            prc_status = 3,
Wei Shoulin's avatar
Wei Shoulin committed
            prc_time = '2021-10-22 11:12:13',
Wei Shoulin's avatar
Wei Shoulin committed
            filename = "CSST_MSC_MS_SCI_20270810142128_20270810142358_100000101_20_img_L1.fits",
            file_path = "/opt/temp/csst/CSST_MSC_MS_SCI_20270810142128_20270810142358_100000101_20_img_L1.fits",
Wei Shoulin's avatar
Wei Shoulin committed
            pipeline_id = "P1",
            refs = {'dark': 1, 'bias': 2, 'flat': 3 })
Wei Shoulin's avatar
c3
Wei Shoulin committed
        print('write:', rec)