def test_loadgecko(self): data = load.load( 'gecko', os.path.join(THIS_DIR, 'input_data', 'test_gecko.gtd')) data = load.load('gecko', [ os.path.join(THIS_DIR, 'input_data', 'test_gecko.gtd'), os.path.join(THIS_DIR, 'input_data', 'test_gecko.gtd') ])
def test_add_pick_badpicknum(self): data = load.load( 'mat', os.path.join(THIS_DIR, 'input_data', 'small_data.mat'))[0] data.picks.add_pick(1) # need to do this to prevent overwriting data.picks.samp1[0, 0] = 1. with self.assertRaises(ValueError): data.picks.add_pick(1)
def main(): """Get arguments, process data, handle saving.""" parser = _get_args() args = parser.parse_args(sys.argv[1:]) if not hasattr(args, 'func'): parser.parse_args(['-h']) radar_data = load(args.ftype, args.fns) if args.name == 'cat': radar_data = concat(radar_data) bn = os.path.splitext(args.fns[0])[0] args.fns = [bn + '.mat'] elif args.name == 'interp': interp(radar_data, **vars(args)) else: for dat in radar_data: args.func(dat, **vars(args)) if args.o is not None: if len(radar_data) > 1: for d, f in zip(radar_data, args.fns): bn = os.path.split(os.path.splitext(f)[0])[1] if bn[-4:] == '_raw': bn = bn[:-4] out_fn = os.path.join(args.o, bn + '_{:s}.mat'.format(args.name)) d.save(out_fn) else: out_fn = args.o radar_data[0].save(out_fn) else: for d, f in zip(radar_data, args.fns): bn = os.path.splitext(f)[0] if bn[-4:] == '_raw': bn = bn[:-4] out_fn = bn + '_{:s}.mat'.format(args.name) d.save(out_fn)
def test_loadbad(self): with self.assertRaises(ValueError): data = load.load( 'bad', os.path.join(THIS_DIR, 'input_data', 'small_data.bad'))
def test_loadpe(self): data = load.load('pe', os.path.join(THIS_DIR, 'input_data', 'test_pe.DT1'))
def test_loadgssi(self): data = load.load('gssi', os.path.join(THIS_DIR, 'input_data', 'test_gssi.DZT'))
def test_loadmat(self): data = load.load( 'mat', os.path.join(THIS_DIR, 'input_data', 'small_data.mat')) self.assertEqual(data[0].data.shape, (20, 40))
def test_segyimporterror(self): # Fail if use segy with self.assertRaises(ImportError): data = load.load('segy', os.path.join(THIS_DIR, 'input_data', 'small_data.segy'))
def test_loadmat(self): # We want normal functionality if not SEGY data = load.load('mat', os.path.join(THIS_DIR, 'input_data', 'small_data.mat')) self.assertEqual(data[0].data.shape, (20, 40))
def main(): parser = _get_args() args = parser.parse_args(sys.argv[1:]) radardata = load.load('mat', [args.fn])[0] pick(radardata, xd=args.xd, yd=args.yd)
def test_loadgprmax(self): data = load.load('gprMax', os.path.join(THIS_DIR, 'input_data', 'bedslope_gprMax_Bscan.h5'))
def test_update_pick_badpicknum(self): data = load.load( 'mat', os.path.join(THIS_DIR, 'input_data', 'small_data.mat'))[0] data.picks.add_pick(1) with self.assertRaises(ValueError): data.picks.update_pick(0, np.zeros((5, data.tnum)))
def test_update_pick(self): data = load.load( 'mat', os.path.join(THIS_DIR, 'input_data', 'small_data.mat'))[0] data.picks.add_pick(1) data.picks.update_pick(1, np.zeros((5, data.tnum))) self.assertTrue(np.all(data.picks.samp1 == 0))
def test_add_pick_overwrite(self): data = load.load( 'mat', os.path.join(THIS_DIR, 'input_data', 'small_data.mat'))[0] data.picks.add_pick(1) data.picks.add_pick(2) self.assertTrue(data.picks.samp1.shape == (1, data.tnum))
def main(): """Get arguments, start picking.""" parser = _get_args() args = parser.parse_args(sys.argv[1:]) radardata = load.load('mat', [args.fn])[0] pick(radardata, xd=args.xd, yd=args.yd)