mci test

csst_dfs_api/common/catalog.py

@@ -49,6 +49,9 @@ class CatalogApi(object):
             return Table()
         fields, dtypes = self._fields_dtypes(query_result.data[0])
         t = Table(names = fields, dtype = dtypes)
+        t.meta['comments'] = [str(query_result.data[0].__class__)]
+        t.meta['total'] = query_result['totalCount']
+
         for rec in query_result.data:
             t.add_row(tuple([rec.__getattribute__(k) for k in fields]))
         return t

tests/mci/test_ifs_cal_merge.py

+import os
+import unittest
+from astropy.io import fits
+
+from csst_dfs_api.mci.calmerge import CalMergeApi
+
+class MCICalMergeApiTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.api = CalMergeApi()
+
+    def test_find(self):
+        recs = self.api.find(detector_no='CCD01',
+            ref_type = "bias",
+            obs_time = ("2021-06-01 11:12:13","2021-06-08 11:12:13"))
+        print('find:', recs)
+
+    def test_get_latest_by_l0(self):
+        rec = self.api.get_latest_by_l0(level0_id='000001102', ref_type = "bias")
+        print('get_latest_by_l0:', rec)
+
+    def test_get(self):
+        rec = self.api.get(id = 2)
+        print('get by id:', rec)
+        rec = self.api.get(cal_id = '2')
+        print('get by cal_id:', rec)
+
+    def test_update_proc_status(self):
+        rec = self.api.update_proc_status(id = 3, status = 1)
+        print('update_proc_status:', rec)
+
+    def test_update_qc1_status(self):
+        rec = self.api.update_qc1_status(id = 3, status = 2)
+        print('update_qc1_status:', rec)    
+
+    def test_write(self):
+        rec = self.api.write(
+            cal_id = '00002',
+            detector_no = '01', 
+            ref_type = "bias",
+            obs_time = "2021-06-04 11:12:13",
+            exp_time = 150,
+            filename = "/opt/dddasd1.params",
+            file_path = "/opt/dddasd1.fits",
+            prc_status = 3,
+            prc_time = '2021-06-04 11:12:13',
+            level0_ids = ['1','2','3','4'])
+        print('write:', rec)
\ No newline at end of file

tests/mci/test_ifs_level0.py

+import os
+import unittest
+from astropy.io import fits
+
+from csst_dfs_api.mci.level0 import Level0DataApi
+
+class MCILevel0DataTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.api = Level0DataApi()
+
+    def test_find(self):
+        recs = self.api.find(obs_id = '000009', obs_type = 'sci', limit = 0)
+        print('find:', recs)
+
+    def test_get(self):
+        rec = self.api.get(id = 100)
+        print('get:', rec)
+
+        rec = self.api.get(level0_id = '1000000102')
+        print('get:', rec)        
+
+    def test_update_proc_status(self):
+        rec = self.api.update_proc_status(level0_id = '000001102', status = 6)
+        print('update_proc_status:', rec)
+
+    def test_update_qc0_status(self):
+        rec = self.api.update_qc0_status(level0_id = '000001102', status = 7)
+        print('update_qc0_status:', rec)
+
+    def test_write(self):
+        rec = self.api.write(
+            level0_id = '000001101',
+            obs_id = '0000011',
+            detector_no = "01",
+            obs_type = "sci",            
+            obs_time = "2021-06-06 11:12:13",
+            exp_time = 150,
+            detector_status_id = 3,
+            filename = "MSC_00001234",
+            file_path = "/opt/MSC_00001234.fits")
+        print('write:', rec)  
\ No newline at end of file

tests/mci/test_ifs_level0_prc.py

+import os
+import unittest
+from astropy.io import fits
+
+from csst_dfs_api.mci.level0prc import Level0PrcApi
+
+class MCILevel0PrcTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.api = Level0PrcApi()
+
+    def test_find(self):
+        recs = self.api.find(level0_id='134')
+        print('find:', recs)
+
+    def test_update_proc_status(self):
+        rec = self.api.update_proc_status(id = 8, status = 4)
+        print('update_proc_status:', rec)
+
+    def test_write(self):
+        rec = self.api.write(
+            level0_id='134', 
+            pipeline_id = "P1",
+            prc_module = "QC0",
+            params_file_path = "/opt/dddasd.params",
+            prc_status = 3,
+            prc_time = '2021-06-04 11:12:13',
+            result_file_path = "/opt/dddasd.header")
+        print('write:', rec)
\ No newline at end of file

tests/mci/test_ifs_level1.py

+import os
+import unittest
+from astropy.io import fits
+
+from csst_dfs_api.mci.level1 import Level1DataApi
+
+class MCILevel1DataTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.api = Level1DataApi()
+
+    def test_find(self):
+        recs = self.api.find(
+            level0_id='000001201',
+            create_time = ("2021-10-01 11:12:13","2021-11-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):
+        rec = self.api.write(
+            level0_id='000001201', 
+            data_type = "sci",
+            cor_sci_id = 2,
+            prc_params = "/opt/dddasd.params",
+            prc_status = 3,
+            prc_time = '2021-06-05 11:12:13',
+            filename = "MSC_MS_210525121500_100000001_09_raw",
+            file_path = "/opt/temp/csst/MSC_MS_210525121500_100000001_09_raw.fits",
+            pipeline_id = "P2",
+            refs = {'dark': 1, 'bias': 2, 'flat': 3, 'sky': 33 })
+        print('write:', rec)
\ No newline at end of file

tests/mci/test_ifs_level1_prc.py

+import os
+import unittest
+from astropy.io import fits
+
+from csst_dfs_api.mci.level1prc import Level1PrcApi
+
+class MCILevel1PrcTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.api = Level1PrcApi()
+
+    def test_find(self):
+        recs = self.api.find(level1_id=1)
+        print('find:', recs)
+
+    def test_update_proc_status(self):
+        rec = self.api.update_proc_status(id = 1, status = 4)
+        print('update_proc_status:', rec)
+
+    def test_write(self):
+        rec = self.api.write(level1_id=1, 
+            pipeline_id = "P1",
+            prc_module = "QC0",
+            params_file_path = "/opt/dddasd.params",
+            prc_status = 3,
+            prc_time = '2021-06-04 11:12:13',
+            result_file_path = "/opt/dddasd.header")
+        print('write:', rec)
\ No newline at end of file