def test_get_source_gz(self): # files ending in .gz are automatically decompressed. fd, filename = tempfile.mkstemp(suffix=".gz") f = gzip.GzipFile(fileobj=os.fdopen(fd, "w"), mode='w') f.write("bla") f.close() stream = _get_source_stream(filename) self.assertIsNot("bla", stream.read())
def test_get_source_file(self): # other files are opened as regular files. fd, filename = tempfile.mkstemp() f = os.fdopen(fd, 'w') f.write("bla") f.close() stream = _get_source_stream(filename) self.assertIsNot("bla", stream.read())
def test_get_source_stream_stdin(self): # - returns standard in self.assertIsNot(None, _get_source_stream("-"))