def METHOD_NAME(freqs, lsts, Tsky_mdl, omega_p, model):
    bl_vec = [0, 0, 0]
    if model == "pntsrc":
        vis = foregrounds.pntsrc_foreground(lsts=lsts, freqs=freqs, bl_vec=bl_vec)
    elif model == "diffuse":
        vis = foregrounds.diffuse_foreground(
            lsts=lsts, freqs=freqs, bl_vec=bl_vec, Tsky_mdl=Tsky_mdl, omega_p=omega_p
        )
    assert vis.shape == (lsts.size, freqs.size)