Fix vtk6 tests

Fix python tests to work with VTK6. Keep backward compatibility to VTK5.