def user_edit(fn, t):

    nmini = 11
    if re.search(r'(specular|varia)', fn):
        nmini = 50

    m = re.search(r'bp\.parse_args\(sim_n=(\d+)', t)
    if not m:
        return t
    n = m.group(1)
    t = re.sub(r"bp\.simargs\['n'\]", f'<%= sm ? {nmini} : {n} %>', t)

    t = re.sub(r'(bp\.parse_args\()sim_n=(\d+),\s*', r'\1', t)
    t = re.sub(r'(bp\.parse_args\()sim_n=(\d+)\)', r'\1)', t)

    if not re.search(r'^#!/usr/bin/env python3\n', t):
        raise Exception("missing shebang in "+fn)
    t = re.sub(r'(^#!.*?\n("""\n(.*\n)+"""\n)?)\n*',
               r'\1'+'<% sm = (e=ENV["SHORTMODE"] and e!="OFF") %>\n', t)

    return t

PYTHONPATH=/usr/lib/python3/dist-packages python3 /usr/lib/python3/dist-packages/yapf -i

applied to Examples/*/*py and Wrap/*/*py

ditto for Spheroid

CMake: change tes target names to avoid duplicate name

def user_edit(fn, t):

    t = re.sub(r'def get_simulation\(sample\):',
               'def get_simulation(sample, **simargs):', t)

    N = 0
    xx = r'\n((\s+?)\w+ = ba\.\w+?Detector)\((\d+), (.+?, .+?), \d+, (.+?, .+?)\)'
    m = re.search(xx, t)
    if m:
        N = int(m.group(3))
    t = re.sub(xx, r"\n\2n = simargs['n']\n\1(n, \4, n, \5)", t)

    xx = r'(\w+ = ba\.\w+?Detector)\((\d+), (.+?, .+?, .+?)\)'
    m = re.search(xx, t)
    if m:
        N = int(m.group(2))
    t = re.sub(xx, r"\1(simargs['n'], \3)", t)

    xx = r"(if __name__ == '__main__':\n)((.*?\n)+?)(\s+\w+ = get_simulation\(\w+)\)\n((.*?\n)*?)\s+bp\.plot_simulation_result\(simulation\.simulate\(\)\)"
    t = re.sub(xx, r"\1" +
               "    plotargs, simargs = bp.kwargs_from_cmdline(sim_n=" + str(N) + ")\n" +
               r"\2" +
               r"\4, **simargs)\n"
               r"\5" +
               "    result = simulation.simulate()\n" +
               "    bp.plot_simulation_result(result, **plotargs)", t)

    return t