Commit 86d62717 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

refactor File.replace_ext

parent 13df314a

csst_common/file.py

+1 −1
Original line number Diff line number Diff line
@@ -6,7 +6,7 @@ class File: 
        self.file_path = file_path

        self.file_path_prefix, self.file_path_ext = os.path.splitext(file_path)



    def switch_extension(self, ext="wht.fits") -> str:

    def replace_ext(self, ext="wht.fits") -> str:

        if ext.startswith("."):

            return self.file_path_prefix + ext

        else:

tests/test_file.py

+2 −2
Original line number Diff line number Diff line
@@ -5,5 +5,5 @@ from csst_common import File 
class TestFile(unittest.TestCase):

    def test_parameterized_module_decorator(self):

        f = File("/path/to/file.fits")

        self.assertEqual(f.switch_extension(".cat"), "/path/to/file.cat")

        self.assertEqual(f.switch_extension("img.cat"), "/path/to/file_img.cat")
 
        self.assertEqual(f.replace_ext(".cat"), "/path/to/file.cat")

        self.assertEqual(f.replace_ext("img.cat"), "/path/to/file_img.cat")