From 25bb1604ed7c81e3192cd7c4616ffdceabd64ad4 Mon Sep 17 00:00:00 2001 From: Michał Górny Date: Sun, 29 Oct 2017 16:39:56 +0100 Subject: compression: Add an explicit function to get suffix --- tests/test_compression.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'tests/test_compression.py') diff --git a/tests/test_compression.py b/tests/test_compression.py index 5f614d3..df78ec8 100644 --- a/tests/test_compression.py +++ b/tests/test_compression.py @@ -658,3 +658,20 @@ class OtherUtilityTests(unittest.TestCase): 'test.lzma', 'test.xz', ])) + + def test_get_compressed_suffix_from_filename(self): + self.assertEqual( + gemato.compression.get_compressed_suffix_from_filename( + 'test.gz'), 'gz') + self.assertEqual( + gemato.compression.get_compressed_suffix_from_filename( + 'test.bz2'), 'bz2') + self.assertEqual( + gemato.compression.get_compressed_suffix_from_filename( + 'test.lzma'), 'lzma') + self.assertEqual( + gemato.compression.get_compressed_suffix_from_filename( + 'test.xz'), 'xz') + self.assertIsNone( + gemato.compression.get_compressed_suffix_from_filename( + 'test')) -- cgit v1.2.3