def __init__(self, fileName): if arsutils.fIsBzipFile(fileName): self.fo = bz2.BZ2File(fileName, "rb", 9) elif arsutils.fIsGzipFile(fileName): self.fo = gzip.GzipFile(fileName, "rb", 9) else: self.fo = open(fileName) self.buf = None self.curPos = 0 self.bufLen = 0 self.curVal = None
def test_fIsBzipFile(self): self.assertEqual(arsutils.fIsBzipFile("me.bz2"), True) self.assertEqual(arsutils.fIsBzipFile("me.bz"), False) self.assertEqual(arsutils.fIsBzipFile("me.bz2.txt"), False)