nir/load_store_vectorizer: Use more imm helpers in the tests.