Skip to content
GitLab
Switch branch/tag
Find file Normal viewHistoryPermalink
test_data_manager.py 2.32 KB
Newer Older
BO ZHANG's avatar
major updates    
BO ZHANG committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50 
# """
# Identifier:     KSC-SJ4-tests/test_data_manager.py
# Name:           test_data_manager.py
# Description:    data manager unit test
# Author:         Bo Zhang
# Created:        2022-09-13
# Modified-History:
#     2022-09-13, Bo Zhang, created
#     2022-09-29, Bo Zhang, added test for CsstMbiDataManager
#     2022-10-28, Bo Zhang, deleted unit test for CsstMsDataManager
# """
# import os
# import unittest
#
# from csst_common.data_manager import CsstMsDataManager
# from csst_common.params import CSST_PARAMS as CP
#
# dir_unittest = "/nfsdata/share/pipeline-unittest/csst_common"
#
#
# class TestCsstMsDataManager(unittest.TestCase):
#     def setUp(self) -> None:
#         self.dm_mbi = CsstMsDataManager.quickstart(
#             ver_sim="C6.2", datatype="mbi", dir_l1=dir_unittest, exposure_id=100)
#         self.dm_sls = CsstMsDataManager.quickstart(
#             ver_sim="C6.2", datatype="sls", dir_l1=dir_unittest, exposure_id=100)
#         self.dm_mbi.target_detectors = None
#         self.dm_sls.target_detectors = None
#
#     def test_mbi_data_existence(self):
#         self.assertTrue(self.dm_mbi.target_detectors == CP["mbi"]["detectors"])
#         self.assertTrue(os.path.exists(self.dm_mbi.l0_detector(6)))
#         self.assertTrue(os.path.exists(self.dm_mbi.l0_log(6)))
#         self.assertTrue(os.path.exists(self.dm_mbi.l0_cat(6)))
#         self.assertTrue(os.path.exists(self.dm_mbi.l0_crs(6)))
#         self.assertTrue(isinstance(self.dm_mbi.l1_detector(6, post="img.fits"), str))
#         self.assertTrue(isinstance(self.dm_mbi.l1_file(pipeline_id="some_file.ext", comment="a demo file"), str))
#
#     def test_sls_data_existence(self):
#         self.assertTrue(self.dm_sls.target_detectors == CP["sls"]["detectors"])
#         self.assertTrue(os.path.exists(self.dm_sls.l0_detector(1)))
#         self.assertTrue(os.path.exists(self.dm_sls.l0_log(1)))
#         self.assertTrue(os.path.exists(self.dm_sls.l0_cat(1)))
#         self.assertTrue(os.path.exists(self.dm_sls.l0_crs(1)))
#         self.assertTrue(isinstance(self.dm_sls.l1_detector(1, post="flt.fits"), str))
#         self.assertTrue(isinstance(self.dm_sls.l1_file(pipeline_id="some_file.ext", comment="a demo file"), str))
#
#     # DFS is not always available
#     # def test_dfs_is_available(self):
#     #     self.assertTrue(self.dm_mbi.dfs_is_available())