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)