test_ifs_sim.py 6.94 KB
Newer Older
Yan Zhaojun's avatar
Yan Zhaojun committed
1
2
3
4
5
6
7
8
9
10
11
12
"""
Identifier:     ifs_sim/tests/test_ifs_sim.py
Name:           test_ifs_sim.py
Description:    Test ifs sim.
Author:         Zhaojun Yan
Created:        2024-04-09
Modified-History:
    2024-04-09, Zhaojun Yan, created

"""

import unittest
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
13
import os
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
14
from csst_ifs_sim import csst_ifs_sim
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
15

Yan Zhaojun's avatar
Yan Zhaojun committed
16
import sys
Yan Zhaojun's avatar
Yan Zhaojun committed
17
18
19
20
21
22

class TestDemoFunction(unittest.TestCase):
    def test_ifs_sim_1(self):
        """
        Aim
        ---
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
23
        Test ifs sim function: SCI case.
Yan Zhaojun's avatar
Yan Zhaojun committed
24
25
26
27
28
29
30
31
32
33
34

        Criteria
        --------
        Pass if the demo function returns `1`.

        Details
        -------
        The demo function returns the length of the input argument list.
        This case aims to test whether the demo function returns `1` if input is `None`.
        """
        # demo function test
Yan Zhaojun's avatar
Yan Zhaojun committed
35
        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
36
        print(dir_path)
Yan Zhaojun's avatar
Yan Zhaojun committed
37
        print(sys.version )
Yan Zhaojun's avatar
Yan Zhaojun committed
38
39
        ###configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C90.config'
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
40
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
41
      
Yan Zhaojun's avatar
Yan Zhaojun committed
42
        sourcein = 'SCI'
Yan Zhaojun's avatar
Yan Zhaojun committed
43
        print(configfile)
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
44
        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
45
46
47
48
49
50
        debug=True 
        
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'no')
        
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
51
        self.assertEqual(
Yan Zhaojun's avatar
Yan Zhaojun committed
52
            1 , 1,
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
53
            "case 1: SCI sim passes.",
Yan Zhaojun's avatar
Yan Zhaojun committed
54
        )
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
55
        
Yan Zhaojun's avatar
Yan Zhaojun committed
56
        ##############################################################
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
    def test_ifs_sim_2(self):
        """
        Aim
        ---
        Test ifs sim function: SCI case.

        Criteria
        --------
        Pass if the demo function returns `1`.

        Details
        -------
        The demo function returns the length of the input argument list.
        This case aims to test whether the demo function returns `1` if input is `None`.
        """
        # demo function test
        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
        print(dir_path)
        print(sys.version )
        ###configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C90.config'
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
77
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
78
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
79
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
80
81
        sourcein = 'BIAS'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
82
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
83
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
84
85
86
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'no')
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
87
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
88
89
90
91
        self.assertEqual(
            1 , 1,
            "case 2:  sim passes.",
        )
Yan Zhaojun's avatar
update    
Yan Zhaojun committed
92
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
        ###################################################################
    def test_ifs_sim_3(self):
        """
        Aim
        ---
        Test ifs sim function: SCI case.

        Criteria
        --------
        Pass if the demo function returns `1`.

        Details
        -------
        The demo function returns the length of the input argument list.
        This case aims to test whether the demo function returns `1` if input is `None`.
        """
        # demo function test
        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
        print(dir_path)
        print(sys.version )
        ###configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
114
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
115
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
116
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
117
118
        sourcein = 'DARK'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
119
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
120
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
121
122
123
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'no')
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
124
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
125
126
127
128
        self.assertEqual(
            1 , 1,
            "case 3:  sim passes.",
        )
Yan Zhaojun's avatar
Yan Zhaojun committed
129
            
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
    ###################################################################
    def test_ifs_sim_4(self):
        """
        Aim
        ---
        Test ifs sim function: SCI case.

        Criteria
        --------
        Pass if the demo function returns `1`.

        Details
        -------
        The demo function returns the length of the input argument list.
        This case aims to test whether the demo function returns `1` if input is `None`.
        """
        # demo function test
        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
        print(dir_path)
        print(sys.version )
        ###configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
151
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
152
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
153
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
154
155
        sourcein = 'LAMP'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
156
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
157
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
158
159
160
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'no')
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
161
162
163
164
        self.assertEqual(
            1 , 1,
            "case 4:  sim passes.",
        )  
Yan Zhaojun's avatar
Yan Zhaojun committed
165
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
    ###################################################################
    def test_ifs_sim_5(self):
        """
        Aim
        ---
        Test ifs sim function: SCI case.

        Criteria
        --------
        Pass if the demo function returns `1`.

        Details
        -------
        The demo function returns the length of the input argument list.
        This case aims to test whether the demo function returns `1` if input is `None`.
        """
        # demo function test
        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
        print(dir_path)
        print(sys.version )
        ###configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
187
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
188
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
189
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
190
191
        sourcein = 'LAMP'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
192
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
193
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
194
195
196
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'yes')
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
197
198
199
200
        self.assertEqual(
            1 , 1,
            "case 5:  sim passes.",
        )
Yan Zhaojun's avatar
Yan Zhaojun committed
201
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
    ###################################################################
    def test_ifs_sim_6(self):
        """
        Aim
        ---
        Test ifs sim function: SCI case.

        Criteria
        --------
        Pass if the demo function returns `1`.

        Details
        -------
        The demo function returns the length of the input argument list.
        This case aims to test whether the demo function returns `1` if input is `None`.
        """
        # demo function test
        dir_path = os.path.join(os.environ['UNIT_TEST_DATA_ROOT'],'ifs_sim/')
        print(dir_path)
        print(sys.version )
        ###configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
223
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
224
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
225
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
226
227
        sourcein = 'FLAT'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
228
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
229
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
230
231
232
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'no')
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
233
234
235
236
        self.assertEqual(
            1 , 1,
            "case 6:  sim passes.",
        )
Yan Zhaojun's avatar
Yan Zhaojun committed
237
238