Commit f78adc77 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 1f9bff72

csst_common/data_manager.py

+0 −1
Original line number Diff line number Diff line
@@ -625,7 +625,6 @@ class CsstMbiDataManager: 
        self.hardcode_history.append(dict(hdcd=fp, comment=comment))

        return fp





    @staticmethod

    def quickstart(ver_sim="C5.2", dir_l1=".", exposure_id=100):

        """ quick dataset generator for tests on dandelion or PMO

tests/test_data_manager.py

+5 −5
Original line number Diff line number Diff line
@@ -47,9 +47,9 @@ class TestCsstMsDataManager(unittest.TestCase): 


    def test_sls_data_existence(self):

        self.assertTrue(self.dm_sls.available_detectors == CP["sls"]["detectors"])

        self.assertTrue(os.path.exists(self.dm_sls.l0_detector(6)))

        self.assertTrue(os.path.exists(self.dm_sls.l0_log(6)))

        self.assertTrue(os.path.exists(self.dm_sls.l0_cat(6)))

        self.assertTrue(os.path.exists(self.dm_sls.l0_crs(6)))

        self.assertTrue(isinstance(self.dm_sls.l1_detector(6, post="img.fits"), str))

        self.assertTrue(os.path.exists(self.dm_sls.l0_detector(1)))

        self.assertTrue(os.path.exists(self.dm_sls.l0_log(1)))

        self.assertTrue(os.path.exists(self.dm_sls.l0_cat(1)))

        self.assertTrue(os.path.exists(self.dm_sls.l0_crs(1)))

        self.assertTrue(isinstance(self.dm_sls.l1_detector(1, post="flt.fits"), str))

        self.assertTrue(isinstance(self.dm_sls.l1_file(name="some_file.ext", comment="a demo file"), str))