Function fix_dft_args

python/simulation.py:72–93

(args, i)

def fix_dft_args(args, i):
    if (
        len(args) > i + 2
        and isinstance(args[i], (int, float))
        and isinstance(args[i + 1], (int, float))
        and isinstance(args[i + 2], int)
    ):
        fcen = args[i]
        df = args[i + 1]
        nfreq = args[i + 2]
        freq = (
            [fcen]
            if nfreq == 1
            else np.linspace(fcen - 0.5 * df, fcen + 0.5 * df, nfreq)
        )
        return args[:i] + (freq,) + args[i + 3 :]
    elif not isinstance(args[i], (np.ndarray, list)):
        raise TypeError(
            "add_dft functions only accept fcen,df,nfreq (3 numbers) or freq (array/list)"
        )
    else:
        return args


def get_num_args(func):