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()