From ec8ee36a89ec11fc2b34d27e2bba8417681e47f2 Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Thu, 22 Sep 2022 16:01:50 +0800 Subject: [PATCH] added unit test for demo function --- csst_proto/some_other_modules.py | 8 +++----- csst_proto/top_level_interface.py | 4 ++-- tests/test_other_functions.py | 11 +++++++++++ 3 files changed, 16 insertions(+), 7 deletions(-) create mode 100644 tests/test_other_functions.py diff --git a/csst_proto/some_other_modules.py b/csst_proto/some_other_modules.py index 426f56e..3c73225 100644 --- a/csst_proto/some_other_modules.py +++ b/csst_proto/some_other_modules.py @@ -1,5 +1,3 @@ -def an_arbitrary_functions(*args): - """ an arbitrary function """ - print("Hi, this is an arbitrary function.") - for _ in args: - print(_) +def a_demo_function(*args): + """ a demo function """ + return None diff --git a/csst_proto/top_level_interface.py b/csst_proto/top_level_interface.py index 471997f..6180569 100644 --- a/csst_proto/top_level_interface.py +++ b/csst_proto/top_level_interface.py @@ -1,5 +1,5 @@ from .flip_image import flip_image, read_test_image -from .some_other_modules import an_arbitrary_functions +from .some_other_modules import a_demo_function -__all__ = ["flip_image", "read_test_image", "an_arbitrary_functions"] +__all__ = ["flip_image", "read_test_image", "a_demo_function"] diff --git a/tests/test_other_functions.py b/tests/test_other_functions.py new file mode 100644 index 0000000..6b77389 --- /dev/null +++ b/tests/test_other_functions.py @@ -0,0 +1,11 @@ +import unittest + +from csst_proto.top_level_interface import a_demo_function + + +class TestCaseDemoFunction(unittest.TestCase): + def test_demo_function(self): + # flip test image + self.assertTrue( + a_demo_function(1) is None + ) -- GitLab