diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_compression.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tests/test_compression.py b/tests/test_compression.py index 0870a1d..e0deaf3 100644 --- a/tests/test_compression.py +++ b/tests/test_compression.py @@ -328,3 +328,26 @@ dGhlIGxhenkgZG9nAADjZCTmHjHqggABLxeBCEmxH7bzfQEAAAAABFla self.assertEqual(xz.read(), TEST_STRING) except gemato.exceptions.UnsupportedCompression: raise unittest.SkipTest('xz compression unsupported') + + +class NoCompressionTest(unittest.TestCase): + """ + Tests for non-compressed data. + """ + + def test_open_potentially_compressed_path(self): + with tempfile.NamedTemporaryFile() as wf: + wf.write(TEST_STRING) + wf.flush() + + with gemato.compression.open_potentially_compressed_path( + wf.name, 'rb') as cf: + self.assertEqual(cf.read(), TEST_STRING) + + def test_open_potentially_compressed_path_write(self): + with tempfile.NamedTemporaryFile() as rf: + with gemato.compression.open_potentially_compressed_path( + rf.name, 'wb') as cf: + cf.write(TEST_STRING) + + self.assertEqual(rf.read(), TEST_STRING) |