Change to new Firedrake `assign` API

[Josh Hope-Collins]

Firedrake changed the API for assign to only accept constants or weighted sums of functions of the same space.
We need to make sure that we only use assign in this way, and update if not.

This change has also broken how we assign to/from the vector functions in the complex proxy space, so that needs to be fixed too.