def METHOD_NAME():
    m = UnitSquareMesh(2, 2)
    f_scalar = Function(FunctionSpace(m, "CG", 1))
    f_vector = Function(VectorFunctionSpace(m, "CG", 1))
    domain = "{[i] : 0 <= i < A.dofs}"
    instructions = "A[i, 0] = 1"
    par_loop((domain, instructions), dx, {'A': (f_vector, WRITE)})
    assert np.allclose(f_vector.dat.data[:, 0], 1.0)
    par_loop((domain, instructions), dx, {'A': (f_scalar, WRITE)})
    assert np.allclose(f_scalar.dat.data, 1.0)