def test_add_pick(self):
#blank pick
self.ip._add_pick()
self.assertTrue(self.ip.dat.picks.samp1.shape[0] == 1)
# Overwrite slot
self.ip._add_pick()
self.assertTrue(self.ip.dat.picks.samp1.shape[0] == 1)
# add to existing
data = RadarData(
os.path.join(THIS_DIR, 'input_data', 'small_data_picks.mat'))
self.ip = InteractivePicker(data)
self.assertTrue(self.ip.dat.picks.samp1.shape[0] == 2)
self.ip._add_pick()
self.assertTrue(self.ip.dat.picks.samp1.shape[0] == 3)
# Check that we can add a non-blank pick
self.ip._add_pick(snum=10, tnum=2)
self.assertTrue(self.ip.dat.picks.samp1.shape[0] == 3)
self.assertTrue(self.ip.current_pick[1, 2] > 5)
# snum but no tnum
self.ip._add_pick(snum=10, tnum=None)
self.assertTrue(self.ip.current_pick[1, 0] > 5)
def test_mode_update(self):
self.ip._mode_update()
self.ip._mode_update()
data = RadarData(
os.path.join(THIS_DIR, 'input_data', 'small_data_picks.mat'))
ip = InteractivePicker(data)
ip._mode_update()
ip._mode_update()
def test_select_lines_click(self):
data = RadarData(
os.path.join(THIS_DIR, 'input_data', 'small_data_picks.mat'))
self.ip = InteractivePicker(data)
event = DummyEvent()
event.artist = self.ip.cline[0]
self.ip._select_lines_click(event)
self.assertEqual(self.ip.pickNumberBox.value(), 1)
event.artist = 'dumdum'
self.ip._select_lines_click(event)
self.assertEqual(self.ip.pickNumberBox.value(), 1)
event.artist = self.ip.cline[1]
self.ip._select_lines_click(event)
self.assertEqual(self.ip.pickNumberBox.value(), 5)
def test_load_cp(self, patchqfd):
patchqfd.getOpenFileName.return_value = ('not_a_file', True)
with self.assertRaises(IOError):
self.ip._load_cp(DummyEvent())
patchqfd.getOpenFileName.return_value = (os.path.join(
THIS_DIR, 'input_data', 'small_data.mat'), True)
with patch('impdar.gui.pickgui.warn') as patchwarn:
self.ip._load_cp(DummyEvent())
self.assertTrue(patchwarn.called)
patchqfd.getOpenFileName.return_value = (os.path.join(
THIS_DIR, 'input_data', 'cross_picked.mat'), True)
self.ip._load_cp(DummyEvent())
data = RadarData(os.path.join(THIS_DIR, 'input_data',
'small_data.mat'))
self.ip = InteractivePicker(data, ydat='depth')
self.ip._load_cp(DummyEvent())
def test_other_lims(self):
data = RadarData(os.path.join(THIS_DIR, 'input_data',
'small_data.mat'))
ip = InteractivePicker(data, xdat='dist')
self.assertEqual(ip.x, 'dist')
with self.assertRaises(ValueError):
ip = InteractivePicker(data, xdat='dum')
ip = InteractivePicker(data, ydat='twtt')
self.assertEqual(ip.y, 'twtt')
ip = InteractivePicker(data, ydat='depth')
self.assertEqual(ip.y, 'depth')
data.nmo_depth = data.travel_time
ip = InteractivePicker(data, ydat='depth')
self.assertEqual(ip.y, 'depth')
with self.assertRaises(ValueError):
ip = InteractivePicker(data, ydat='dum')
with self.assertRaises(ValueError):
ip = InteractivePicker(data, ydat='elev')
data.elevation = np.arange(ip.dat.tnum)
data.flags.elev = True
ip = InteractivePicker(data, x_range=None)
self.assertEqual(ip.x_range, (0, ip.dat.tnum))
def setUp(self):
data = RadarData(os.path.join(THIS_DIR, 'input_data',
'small_data.mat'))
self.ip = InteractivePicker(data)