def test_plot(self):
"""
Test for plotting functions
"""
try:
import matplotlib.pyplot as plt
except Exception:
return True
# step_func
tlist = np.linspace(0., 2 * np.pi, 20)
processor = Processor(N=1, spline_kind="step_func")
processor.add_control(sigmaz(), label="sz")
processor.set_all_coeffs({"sz": np.array([np.sin(t) for t in tlist])})
processor.set_all_tlist(tlist)
fig, _ = processor.plot_pulses(use_control_latex=False)
# testing under Xvfb with pytest-xvfb complains if figure windows are
# left open, so we politely close it:
plt.close(fig)
# cubic spline
tlist = np.linspace(0., 2 * np.pi, 20)
processor = Processor(N=1, spline_kind="cubic")
processor.add_control(sigmaz(), label="sz")
processor.set_all_coeffs({"sz": np.array([np.sin(t) for t in tlist])})
processor.set_all_tlist(tlist)
fig, _ = processor.plot_pulses(use_control_latex=False)
# testing under Xvfb with pytest-xvfb complains if figure windows are
# left open, so we politely close it:
plt.close(fig)
def testPlot(self):
"""
Test for plotting functions
"""
try:
import matplotlib.pyplot as plt
except Exception:
return True
# step_func
tlist = np.linspace(0., 2*np.pi, 20)
processor = Processor(N=1, spline_kind="step_func")
processor.add_control(sigmaz())
processor.pulses[0].tlist = tlist
processor.pulses[0].coeff = np.array([np.sin(t) for t in tlist])
processor.plot_pulses()
plt.clf()
# cubic spline
tlist = np.linspace(0., 2*np.pi, 20)
processor = Processor(N=1, spline_kind="cubic")
processor.add_control(sigmaz())
processor.pulses[0].tlist = tlist
processor.pulses[0].coeff = np.array([np.sin(t) for t in tlist])
processor.plot_pulses()
plt.clf()