def testReadFile(self): data = [ (b"\xef\xbb\xbfFoo", "Foo"), (b"Bar", "Bar"), ] for src, expected in data: fl = BytesIO(src) dst = readFile(fl) self.assertEqual(dst, expected)
def convert(self, filename): ext = os.path.splitext(filename)[1] if ext == '.gz': fl = gzip.open(filename, 'rb') else: fl = open(filename, 'rb') src = readFile(fl) src = skipHeader(src) return self._doConvert(src)