From 5e2fd0cc368515b84d3160ef66cbcad158b2d128 Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Thu, 29 Sep 2022 16:57:02 +0800 Subject: [PATCH] added unit test for CsstMbiDataManager --- tests/test_data_manager.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/tests/test_data_manager.py b/tests/test_data_manager.py index 1d96c6b..1ea3f85 100644 --- a/tests/test_data_manager.py +++ b/tests/test_data_manager.py @@ -6,12 +6,27 @@ Author: Bo Zhang Created: 2022-09-13 Modified-History: 2022-09-13, Bo Zhang, created - 2022-09-13, Bo Zhang, fixed a bug + 2022-09-29, Bo Zhang, added test for CsstMbiDataManager """ +import os import unittest from csst_common.data_manager import CsstMbiDataManager +class TestDataManager(unittest.TestCase): + def setUp(self) -> None: + self.dm = CsstMbiDataManager.quickstart(ver_sim="C5.2", dir_l1=".", exposure_id=100) + + def test_l0data_existence(self): + self.assertTrue(self.dm.available_detectors == 18) + self.assertTrue(os.path.exists(self.dm.l0_detector(6))) + self.assertTrue(os.path.exists(self.dm.l0_log(6))) + self.assertTrue(os.path.exists(self.dm.l0_cat(6))) + self.assertTrue(os.path.exists(self.dm.l0_crs(6))) + self.assertTrue(isinstance(self.dm.l1_detector(6, post="img.fits"), str)) + self.assertTrue(isinstance(self.dm.l1_file(name="some_file.ext", comment="a demo file"), str)) + + class TestParams(unittest.TestCase): def test_params(self): self.assertTrue(True) -- GitLab