def test_arr2vtkIdList(self): """Test array to vtkIdList conversion.""" a = [1, 2, 3, 4, 5] p = array_handler.array2vtkIdList(a) for i, j in enumerate(a): self.assertEqual(p.GetId(i), j) p = vtk.vtkIdList() ident = id(p) p = array_handler.array2vtkIdList(numpy.array(a), p) for i, j in enumerate(a): self.assertEqual(p.GetId(i), j) self.assertEqual(id(p), ident) self.assertRaises(AssertionError, array_handler.array2vtkIdList, [[1, 2, 3]])