Modify python test for VTK 5.