def test_limits_trace_curves_to_xy_edges(self, diagonal_trace): spt = SpectralTrace(diagonal_trace) xy_edges = {"x_min": -25, "x_max": -15, "y_min": 10, "y_max": 20} mtcs_all = spt.get_trace_curves(0.015) mtcs_xy_limited = spt.get_trace_curves(0.015, xy_edges=xy_edges) assert len(mtcs_all) > len(mtcs_xy_limited) if PLOTS: for mtc in mtcs_xy_limited: plt.plot(mtc.x, mtc.y) spt.plot(spt.wave_min, spt.wave_max) plt.axhline(xy_edges["y_min"]) plt.axhline(xy_edges["y_max"]) plt.axvline(xy_edges["x_min"]) plt.axvline(xy_edges["x_max"]) plt.show()
def test_plots(self, curved_trace): spt = SpectralTrace(curved_trace) if PLOTS: spt.plot(0.5, 2.5) plt.show()