can you put the test in the test method? I think that's preferable so it's not split up.