Refactor PME tests for better usability