import numpy as np
import modes as ms

widths = np.arange(0.5, 2.0, 0.5)
wgs = [ms.waveguide(width=width) for width in widths]
r = ms.sweep_waveguide(
    wgs, widths, x_label="waveguide width", fraction_mode_list=[1, 2],
)
print(r["n_effs"][0])