Skip to content
GitLab
An error occurred while loading the file. Please try again.
Switch branch/tag
Find file BlameHistoryPermalink
test_level2.py 3.02 KiB
Open in Web IDE
1 
import unittest
2
3 
from csst_dfs_client import level2
4
5 
class Level2TestCase(unittest.TestCase):
6
7 
    def setUp(self):
8 
        pass
9
10 
    def test_find(self):
11 
        result = level2.find(obs_id = "1010940063886712")
12 
        print(result)
13 
        self.assertEqual(result.code, 200, "error code: " + str(result.code))
14 
        self.assertIsNotNone(result.data, "error message: " + result.message)
15
16 
    def test_find_by_level2_id(self):
17 
        result = level2.find_by_level2_id(level2_id = "1010940063886712")
18 
        print(result)
19 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
20 
        self.assertIsNotNone(result.data, "error message: " + result.message)
21
22 
    def test_update_qc2_status(self):
23 
        result = level2.update_qc2_status(level2_id = "1010940063886712", data_type='csst-msc-l2-mbi-cat', qc2_status=1)
24 
        print(result)
25 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
26
27 
    def test_update_prc_status(self):
28 
        result = level2.update_prc_status(level2_id = "1010940063886712", data_type='csst-msc-l2-mbi-cat', prc_status=3)
29 
        print(result)
30 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
31
32 
    def test_update_qc2_status_by_file_name(self):
33 
        result = level2.update_qc2_status_by_file_name(file_name = "CSST_MSC_MS_SCI_20310423084104_20310423084334_10109400638867_12_L2_V01_CATMIX.fits", qc2_status=1)
34 
        print(result)
35 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
36
37 
    def test_update_prc_status_by_file_name(self):
38 
        result = level2.update_prc_status_by_file_name(file_name = "CSST_MSC_MS_SCI_20310423084104_20310423084334_10109400638867_12_L2_V01_CATMIX.fits", prc_status=3)
39 
        print(result)
40 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
41
42 
    def test_write(self):
43 
        file_path = "/Users/wsl/temp/csst/import/CSST_MSC_MS_SCI_20310423084104_20310423084334_10109400638867_12_L2_V01_CATMIX.fits"
44 
        result = level2.write(local_file = file_path,
45 
            module_id = "MSC",
46 
            level0_id="1010940063886712",
47 
            level1_id = "1010940063886712",
48 
            level2_id = "1010940063886712",
49 
            data_type='csst-msc-l2-mbi-cat',
50 
            file_name = "CSST_MSC_MS_SCI_20310423084104_20310423084334_10109400638867_12_L2_V01_CATMIX.fits",
51 
            brick_id = 254,
52 
            qc2_status=12,
53 
            pipeline_id = "csst-msc-l2-mbi",
54 
            build=1,
55 
            dataset="094",
56 
            prc_status=1)
57 
        print(result)
58 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
59
60 
    def test_catalog_query(self):
61 
        result = level2.catalog_query(sql="select * from csst_msc_l2_mbi_cat")
62 
        print(result)
63 
        self.assertEqual(result.code, 200, "error code: " + str(result.code) + ", message: " + result.message)
64