Why not just set this up in the specific test method? `if` statements in tests are always a bad sign from my experience.