TODO get expansion condesion dolfin func - np array straight