TST: Refactor plot test to put repetitive code in helper methods