tweak to python plot tests