Fixed visualisation method tests so that they work with python 2.6