From ab15c00f59f44dda752b6a8ca212cb5ad112d4ec Mon Sep 17 00:00:00 2001 From: zhangxin Date: Sat, 20 Apr 2024 12:53:55 +0800 Subject: [PATCH] modify unitest root --- tests/test_SpecDisperse.py | 10 ++++++++-- tests/test_Straylight.py | 2 +- tests/{det_effect_unit_test.py => test_effect_unit.py} | 2 +- 3 files changed, 10 insertions(+), 4 deletions(-) rename tests/{det_effect_unit_test.py => test_effect_unit.py} (99%) diff --git a/tests/test_SpecDisperse.py b/tests/test_SpecDisperse.py index 0931b52..2b9529d 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 c79dc6f..ea6de76 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 92cd481..d77c9b0 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) -- GitLab