def test_singular_values_plot_mpl_superimpose_nyq(ss_mimo_ct, ss_mimo_dt):
sys_ct = ss_mimo_ct.sys
sys_dt = ss_mimo_dt.sys
omega_all = np.logspace(-3, 2, 1000)
plt.figure()
singular_values_plot(sys_ct, omega_all, plot=True)
singular_values_plot(sys_dt, omega_all, plot=True)
fig = plt.gcf()
allaxes = fig.get_axes()
assert (len(allaxes) == 1)
assert (allaxes[0].get_label() == 'control-sigma')
nyquist_line = allaxes[0].lines[-1].get_data()
assert (len(nyquist_line[0]) == 2)
assert (nyquist_line[0][0] == nyquist_line[0][1])
assert (nyquist_line[0][0] == np.pi / sys_dt.dt)
def test_singular_values_plot_mpl_base(ss_mimo_ct, ss_mimo_dt):
sys_ct = ss_mimo_ct.sys
sys_dt = ss_mimo_dt.sys
plt.figure()
singular_values_plot(sys_ct, plot=True)
fig = plt.gcf()
allaxes = fig.get_axes()
assert (len(allaxes) == 1)
assert (allaxes[0].get_label() == 'control-sigma')
plt.figure()
singular_values_plot([sys_ct, sys_dt],
plot=True,
Hz=True,
dB=True,
grid=False)
fig = plt.gcf()
allaxes = fig.get_axes()
assert (len(allaxes) == 1)
assert (allaxes[0].get_label() == 'control-sigma')
def test_singular_values_plot(tsystem):
sys = tsystem.sys
for omega_ref, sigma_ref in zip(tsystem.omegas, tsystem.sigmas):
sigma, _ = singular_values_plot(sys, omega_ref, plot=False)
np.testing.assert_almost_equal(sigma, sigma_ref)