Example #1
0
 def test_openinput(self):
     """Test Baker.openinput()"""
     self.assertTrue(baker.openinput('-') is sys.stdin)
     tempdir = tempfile.mkdtemp()
     for ext, opener in [(".gz", gzip.GzipFile), (".bz2", bz2.BZ2File)]:
         g = os.path.join(tempdir, "test" + ext)
         input = TestBaker.bytes(INPUT_TEST, 'utf-8')
         fobj = opener(g, "w")
         fobj.write(input)
         fobj.close()
         self.assertEqual(baker.openinput(g).read(), input)