Update px pylab test to match new output of pylab

This is similar to the fix in #3722 but only exposed when running the tests with ```--all```