import numpy as np

from symop.modes.envelopes import GaussianEnvelope
import symop.viz as viz

env = GaussianEnvelope(
    omega0=10.0,
    sigma=1.0,
    tau=0.0,
    phi0=0.0,
)


viz.plot(
    env,
    show_freq=True,
    normalize_spectrum=True,
    freq_relative=True,
    title="Gaussian envelope",
    show=False,
)