def test_wrap_vtk(self): """Test if the wrap_vtk function works.""" o = vtk.vtkVolume() w = tvtk_helper.wrap_vtk(o) self.assertEqual(w.__class__.__name__, 'Volume') w1 = tvtk_helper.wrap_vtk(w) self.assertEqual(w, w1) del w1, w, o class A: pass a = A() w = tvtk_helper.wrap_vtk(a) self.assertEqual(a, w)