示例#1
0
 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())
示例#2
0
 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())
示例#3
0
 def test_get_source_stream_stdin(self):
     # - returns standard in
     self.assertIsNot(None, _get_source_stream("-"))