diff --git a/tests/test_SpecDisperse.py b/tests/test_SpecDisperse.py index 0931b5212756c0f14c5c7ca8e3610ba14359d74e..2b9529d86724db8b4d35b5b72845dee52bd98a84 100644 --- a/tests/test_SpecDisperse.py +++ b/tests/test_SpecDisperse.py @@ -145,16 +145,22 @@ class TestSpecDisperse(unittest.TestCase): def __init__(self, methodName='runTest'): super(TestSpecDisperse,self).__init__(methodName) - self.filePath('csst_fz_gc0') + self.filePath('csst_msc_sim/test_sls_and_straylight') # self.conff = conff # self.throughputf = throughputf + def filePath(self, file_name): fn = os.path.join(os.getenv('UNIT_TEST_DATA_ROOT'), file_name) self.conff= os.path.join(fn, 'CSST_GI2.conf') self.throughputf= os.path.join(fn, 'GI.Throughput.1st.fits') self.testDir = fn + self.outDataFn = os.path.join(fn,'output') + if os.path.isdir(self.outDataFn): + pass + else: + os.mkdir(self.outDataFn) def test_rotate901(self): m = np.array([[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]]) @@ -484,7 +490,7 @@ class TestSpecDisperse(unittest.TestCase): print('Spec double disperse test') from astropy.io import fits - fits.writeto('test.fits',chip.img.array, overwrite = True) + fits.writeto(os.path.join(self.outDataFn,'test_sls_doubleDisp.fits'),chip.img.array, overwrite = True) # plt.figure() # plt.imshow(chip.img.array) diff --git a/tests/test_Straylight.py b/tests/test_Straylight.py index c79dc6f7980791dc80bee79eed12bf1038cdb3fe..ea6de764327c3968c8d11fb3998019d85b99ab62 100644 --- a/tests/test_Straylight.py +++ b/tests/test_Straylight.py @@ -74,7 +74,7 @@ class TestStraylight(unittest.TestCase): # print(file_name) # fn = os.path.join(os.getenv('UNIT_TEST_DATA_ROOT'), file_name) # self.pointingData = np.loadtxt(os.path.join(fn, 'Straylight_test.dat'), dtype=np.double) - self.filePath('csst_fz_gc0') + self.filePath('csst_msc_sim/test_sls_and_straylight') self.filter = filter self.grating = grating diff --git a/tests/det_effect_unit_test.py b/tests/test_effect_unit.py similarity index 99% rename from tests/det_effect_unit_test.py rename to tests/test_effect_unit.py index 92cd4819c06b581361d7626aa43fe9706b5518d8..d77c9b0bd84e8b0faeebb90270e953be9005ecf2 100644 --- a/tests/det_effect_unit_test.py +++ b/tests/test_effect_unit.py @@ -17,7 +17,7 @@ class DetTest(unittest.TestCase): def __init__(self, methodName='runTest'): super(DetTest,self).__init__(methodName) - self.filePath('csst_fz_gc0') + self.filePath('csst_msc_sim/test_sls_and_straylight') def filePath(self, file_name): self.datafn = os.path.join(os.getenv('UNIT_TEST_DATA_ROOT'), file_name)