def METHOD_NAME(u):
    ag = u.atoms[2:5]
    ref = np.array([cog(u, ag, i)
                    for i in range(3)])
    p = multiprocessing.Pool(2)
    res = np.array([p.apply(cog, args=(u, ag, i))
                    for i in range(3)])
    p.close()
    assert_equal(ref, res)