[firedrake] Restricting a test function space