test_ifs_sim.py 6.97 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
        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
159
160
161
        result_path=dir_path+'ifs_sim_result'
        
        csst_ifs_sim.runIFSsim(sourcein,  configfile, dir_path, result_path, 1, debug,'no')
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
162
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
163
164
165
166
        self.assertEqual(
            1 , 1,
            "case 4:  sim passes.",
        )  
Yan Zhaojun's avatar
Yan Zhaojun committed
167
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
    ###################################################################
    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
189
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
190
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
191
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
192
193
        sourcein = 'LAMP'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
194
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
195
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
196
197
198
        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
199
200
201
202
        self.assertEqual(
            1 , 1,
            "case 5:  sim passes.",
        )
Yan Zhaojun's avatar
Yan Zhaojun committed
203
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
    ###################################################################
    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
225
        
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
226
        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C90.config'
Yan Zhaojun's avatar
Yan Zhaojun committed
227
      
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
228
229
        sourcein = 'FLAT'
        print(configfile)
Yan Zhaojun's avatar
Yan Zhaojun committed
230
                
Yan Zhaojun's avatar
debug    
Yan Zhaojun committed
231
        debug=True        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
232
        
Yan Zhaojun's avatar
test    
Yan Zhaojun committed
233
234
235
        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
236
237
238
239
        self.assertEqual(
            1 , 1,
            "case 6:  sim passes.",
        )
Yan Zhaojun's avatar
Yan Zhaojun committed
240
241