def test_plotPLOTARGS(self, mock_plot_rad, mock_show): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')]) mock_plot_rad.assert_called_with(Any(RadarData), xdat='tnum', ydat='twtt', x_range=None, pick_colors=None, clims=None, cmap=Any(object), flatten_layer=None) mock_plot_rad.reset_called() plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], xd=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='dist', ydat='twtt', x_range=None, pick_colors=None, clims=None, cmap=Any(object), flatten_layer=None) mock_plot_rad.reset_called() plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], yd=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='tnum', ydat='depth', x_range=None, pick_colors=None, clims=None, cmap=Any(object), flatten_layer=None) mock_plot_rad.reset_called() plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], xd=True, yd=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='dist', ydat='depth', x_range=None, pick_colors=None, clims=None, cmap=Any(object), flatten_layer=None) mock_plot_rad.reset_called() # Check that we can save plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], xd=True, yd=True, s=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='dist', ydat='depth', x_range=None, pick_colors=None, clims=None, cmap=Any(object), flatten_layer=None) mock_plot_rad.reset_called()
def test_plotLOADPE(self, mock_plot_rad): plot.plot([os.path.join(THIS_DIR, 'input_data', 'test_pe.DT1')], filetype='pe') mock_plot_rad.assert_called_with(Any(RadarData), xdat='tnum', ydat='twtt', x_range=None)
def plot_radargram(fns=None, s=False, o=None, xd=False, yd=False, o_fmt='png', dpi=300, in_fmt='mat', picks=False, clims=None, cmap='gray', flatten_layer=None, **kwargs): """Plot data as a radio echogram.""" plot.plot(fns, xd=xd, yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt, pick_colors=picks, cmap=cmap, clims=clims, flatten_layer=flatten_layer)
def test_plotPLOTSPECDENSE(self, mock_plot_specdense, mock_show): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], spectra=(0, 1), window=0, scaling=1) mock_plot_specdense.assert_called_with(Any(RadarData), (0, 1), window=0, scaling=1)
def test_plotLOADPE(self, mock_plot_rad, mock_show): plot.plot([os.path.join(THIS_DIR, 'input_data', 'test_pe.DT1')], filetype='pe') mock_plot_rad.assert_called_with(Any(RadarData), xdat='tnum', ydat='twtt', x_range=None, pick_colors=None, clims=None, cmap=Any(object), flatten_layer=None)
def plot_power(fns=None, layer=None, s=False, o=None, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): plot.plot(fns, power=layer, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt)
def plot_power(fns=None, layer=None, s=False, o=None, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): """Plot the return power of a particular layer.""" plot.plot(fns, power=layer, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt)
def plot_radargram(fns=None, s=False, o=None, xd=False, yd=False, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): plot.plot(fns, xd=xd, yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt)
def plot_traces(fns=None, t_start=None, t_end=None, yd=False, s=False, o=None, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): plot.plot(fns, tr=(t_start, t_end), yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt)
def plot_traces(fns=None, t_start=None, t_end=None, yd=False, s=False, o=None, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): """Plot traces in terms of amplitude vs some vertical variable.""" plot.plot(fns, tr=(t_start, t_end), yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt)
def test_plotPLOTARGS(self, mock_plot_rad): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')]) mock_plot_rad.assert_called_with(Any(RadarData), xdat='tnum', ydat='twtt', x_range=None) mock_plot_rad.reset_called() plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], xd=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='dist', ydat='twtt', x_range=None) mock_plot_rad.reset_called() plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], yd=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='tnum', ydat='depth', x_range=None) mock_plot_rad.reset_called() plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], xd=True, yd=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='dist', ydat='depth', x_range=None) mock_plot_rad.reset_called() # Check that we can save plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], xd=True, yd=True, s=True) mock_plot_rad.assert_called_with(Any(RadarData), xdat='dist', ydat='depth', x_range=None) mock_plot_rad.reset_called()
def plot_spectrogram(fns=None, freq_lower=None, freq_upper=None, window=None, scaling='spectrum', yd=False, s=False, o=None, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): """Plot a spectrogram.""" plot.plot(fns, spectra=(freq_lower, freq_upper), window=window, scaling=scaling, yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt)
def plot_hft(fns=None, s=False, o=None, xd=False, yd=False, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): """ Plot the fourier spectrum of the data in the horizontal. Might be useful for guessing how to horizontally filter. """ plot.plot(fns, xd=xd, yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt, hft=True)
def plot_ft(fns=None, s=False, o=None, xd=False, yd=False, o_fmt='png', dpi=300, in_fmt='mat', **kwargs): """ Plot the fourier spectrum of the data. Can be useful if you have mystery data of unknown frequency. """ plot.plot(fns, xd=xd, yd=yd, s=s, o=o, ftype=o_fmt, dpi=dpi, filetype=in_fmt, ft=True)
def test_plotHFT(self, mock_plot_hft, mock_show): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], hft=True) mock_plot_hft.assert_called_with(Any(RadarData))
def test_plotPLOTPOWER(self, mock_plot_power): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], power=0) mock_plot_power.assert_called_with(Any(RadarData), 0)
def test_plotPLOTTRACES(self, mock_plot_tr): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], tr=0) mock_plot_tr.assert_called_with(Any(RadarData), 0, ydat='twtt')
def test_plotBADINPUT(self): with self.assertRaises(ValueError): plot.plot([os.path.join(THIS_DIR, 'input_data', 'small_data.mat')], tr=0, power=1)