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)
