[test][python][cleanup] Make initialization order more sensible

The old one wasn't wrong but it seems to make more sense and easier to read
if the gridvariables are initialized before passing them to the assembler.