Add a few missing assertions

Note that some of them were already set by Kokkos (in debug mode), but
it seems better to only rely on `pugs` definitions