def test_fileZipMode(self): filemode1, zipmode1 = fileZipMode('20120115.13401954.bhz.pkl') self.assertEqual(filemode1, 'pkl') self.assertEqual(zipmode1, None) filemode2, zipmode2 = fileZipMode('20120115.13401954.bhz.pkl.gz') self.assertEqual(filemode2, 'pkl') self.assertEqual(zipmode2, 'gz') filemode3, zipmode3 = fileZipMode('TA.SAC') self.assertEqual(filemode3, 'sac') self.assertEqual(zipmode3, None)
def main(): opts, ifiles = getOptions() if opts.s2p: print('File conversion: sac --> pkl') sac2pkl(ifiles, opts.ofilename, opts.delta, opts.zipmode) elif opts.p2s: print('File conversion: pkl --> sac') filemode, zipmode = fileZipMode(ifiles[0]) for pkfile in ifiles: pkl2sac(pkfile, zipmode)