Commit 1e4d9676 authored by Yan Zhaojun's avatar Yan Zhaojun
Browse files

Edit test_ifs_sim.py

parent d24f868e

tests/test_ifs_sim.py

+32 −19
Original line number Diff line number Diff line
@@ -11,7 +11,7 @@ Modified-History: 


import unittest

import os

from csst_ifs_sim import csst_ifs_sim

from csst_ifs_sim import csst_ifs_sim as ifs_sim



import sys
@@ -40,7 +40,14 @@ class TestDemoFunction(unittest.TestCase): 
        print(sys.version)

        # ##configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C10.config'



        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10.config'

        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10_json.config'

        

        # read json file ########

        # #

        import json

        jsonfile = os.path.join(dir_path, 'IFS_inputdata/configData/ngc6217_pzcs_xn.json')

        with open (jsonfile ,'r') as file:            

            json_data=json.load(file)



        sourcein = 'SCI'

        print(configfile)
@@ -49,8 +56,9 @@ class TestDemoFunction(unittest.TestCase): 


        # result_path = dir_path+'ifs_sim_result'

        result_path = os.path.join(dir_path, 'ifs_sim_result')

        csst_ifs_sim.runIFSsim(sourcein, configfile,

                               dir_path, result_path, 1, debug, 'no')



        ifs_sim.runIFSsim(json_data, sourcein, configfile,

                          dir_path, result_path, 1, debug, applyhole)



        self.assertEqual(

            1, 1,
@@ -79,7 +87,8 @@ class TestDemoFunction(unittest.TestCase): 
        print(sys.version)

        # ##configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C10.config'



        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10.config'

        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10_json.config'

        json_data=[]



        sourcein = 'BIAS'

        print(configfile)
@@ -87,8 +96,8 @@ class TestDemoFunction(unittest.TestCase): 
        debug = True

        result_path = dir_path+'ifs_sim_result'



        csst_ifs_sim.runIFSsim(sourcein,  configfile,

                               dir_path, result_path, 1, debug, 'no')

        ifs_sim.runIFSsim(json_data, sourcein, configfile,

                          dir_path, result_path, 1, debug, applyhole)



        self.assertEqual(

            1, 1,
@@ -117,7 +126,8 @@ class TestDemoFunction(unittest.TestCase): 
        print(sys.version)

        # ##configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C10.config'



        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10.config'

        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10_json.config'

        json_data=[]



        sourcein = 'DARK'

        print(configfile)
@@ -125,8 +135,8 @@ class TestDemoFunction(unittest.TestCase): 
        debug = True

        result_path = dir_path+'ifs_sim_result'



        csst_ifs_sim.runIFSsim(sourcein,  configfile,

                               dir_path, result_path, 1, debug, 'no')

        ifs_sim.runIFSsim(json_data, sourcein, configfile,

                          dir_path, result_path, 1, debug, applyhole)



        self.assertEqual(

            1, 1,
@@ -155,7 +165,8 @@ class TestDemoFunction(unittest.TestCase): 
        print(sys.version)

        # ##configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C10.config'



        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10.config'

        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10_json.config'

        json_data=[]



        sourcein = 'LAMP'

        print(configfile)
@@ -164,8 +175,8 @@ class TestDemoFunction(unittest.TestCase): 


        result_path = dir_path+'ifs_sim_result'



        csst_ifs_sim.runIFSsim(sourcein,  configfile,

                               dir_path, result_path, 1, debug, 'no')

        ifs_sim.runIFSsim(json_data, sourcein, configfile,

                          dir_path, result_path, 1, debug, applyhole)



        self.assertEqual(

            1, 1,
@@ -194,7 +205,8 @@ class TestDemoFunction(unittest.TestCase): 
        print(sys.version)

        # ##configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C10.config'



        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10.config'

        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10_json.config'

        json_data=[]



        sourcein = 'LAMP'

        print(configfile)
@@ -202,8 +214,8 @@ class TestDemoFunction(unittest.TestCase): 
        debug = True

        result_path = dir_path+'ifs_sim_result'



        csst_ifs_sim.runIFSsim(sourcein,  configfile,

                               dir_path, result_path, 1, debug, 'yes')

        ifs_sim.runIFSsim(json_data, sourcein, configfile,

                          dir_path, result_path, 1, debug, applyhole)

        self.assertEqual(

            1, 1,

            "case 5:  sim passes.",
@@ -231,7 +243,8 @@ class TestDemoFunction(unittest.TestCase): 
        print(sys.version)

        # ##configfile = dir_path+'IFS_inputdata/configData/IFS_sim_C10.config'



        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10.config'

        configfile = './csst_ifs_sim/ifs_data/IFS_sim_C10_json.config'

        json_data=[]



        sourcein = 'FLAT'

        print(configfile)
@@ -240,8 +253,8 @@ class TestDemoFunction(unittest.TestCase): 


        result_path = dir_path+'ifs_sim_result'



        csst_ifs_sim.runIFSsim(sourcein,  configfile,

                               dir_path, result_path, 1, debug, 'no')

        ifs_sim.runIFSsim(json_data, sourcein, configfile,

                          dir_path, result_path, 1, debug, applyhole)

        self.assertEqual(

            1, 1,

            "case 6:  sim passes.",