nir: Don't replace the whole shader in NIR_TEST_*

Instead, we replace everything except the nir_shader itself which we leave in place. This makes the NIR_TEST_* variables work with gallium drivers.