def test_arr2cell_array(self): """Test Numeric array to vtkCellArray conversion.""" # Test list of lists. a = [[0], [1, 2], [3, 4, 5], [6, 7, 8, 9]] cells = array_handler.array2vtkCellArray(a) z = numpy.array([1, 0, 2, 1,2, 3, 3,4,5, 4, 6,7,8,9]) arr = array_handler.vtk2array(cells.GetData()) self.assertEqual(numpy.sum(arr - z), 0) self.assertEqual(len(arr.shape), 1) self.assertEqual(len(arr), 14) # Test if optional argument stuff also works. cells = vtk.vtkCellArray() ident = id(cells) cells = array_handler.array2vtkCellArray(a, cells) self.assertEqual(id(cells), ident) arr = array_handler.vtk2array(cells.GetData()) self.assertEqual(numpy.sum(arr - z), 0) self.assertEqual(cells.GetNumberOfCells(), 4) # Make sure this resets the cell array and does not add to the # existing list! cells = array_handler.array2vtkCellArray(a, cells) self.assertEqual(cells.GetNumberOfCells(), 4) # Test Numeric array handling. N = 3 a = numpy.zeros((N,3), numpy.int) a[:,1] = 1 a[:,2] = 2 cells = array_handler.array2vtkCellArray(a) arr = array_handler.vtk2array(cells.GetData()) expect = numpy.array([3, 0, 1, 2]*3, numpy.int) self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)), True) self.assertEqual(cells.GetNumberOfCells(), N) # Test if a list of Numeric arrays of different cell lengths works. l_a = [a[:,:1], a, a[:2,:2]] cells = array_handler.array2vtkCellArray(l_a) arr = array_handler.vtk2array(cells.GetData()) expect = numpy.array([1, 0]*3 + [3, 0, 1, 2]*3 + [2, 0,1]*2, numpy.int) self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)), True) self.assertEqual(cells.GetNumberOfCells(), N*2 + 2) # This should not take a long while. This merely tests if a # million cells can be created rapidly. N = int(1e6) a = numpy.zeros((N,3), numpy.int) a[:,1] = 1 a[:,2] = 2 cells = array_handler.array2vtkCellArray(a) self.assertEqual(cells.GetNumberOfCells(), N)
def test_arr2cell_array(self): """Test Numeric array to vtkCellArray conversion.""" # Test list of lists. a = [[0], [1, 2], [3, 4, 5], [6, 7, 8, 9]] cells = array_handler.array2vtkCellArray(a) z = numpy.array([1, 0, 2, 1, 2, 3, 3, 4, 5, 4, 6, 7, 8, 9]) arr = array_handler.vtk2array(cells.GetData()) self.assertEqual(numpy.sum(arr - z), 0) self.assertEqual(len(arr.shape), 1) self.assertEqual(len(arr), 14) # Test if optional argument stuff also works. cells = vtk.vtkCellArray() ident = id(cells) cells = array_handler.array2vtkCellArray(a, cells) self.assertEqual(id(cells), ident) arr = array_handler.vtk2array(cells.GetData()) self.assertEqual(numpy.sum(arr - z), 0) self.assertEqual(cells.GetNumberOfCells(), 4) # Make sure this resets the cell array and does not add to the # existing list! cells = array_handler.array2vtkCellArray(a, cells) self.assertEqual(cells.GetNumberOfCells(), 4) # Test Numeric array handling. N = 3 a = numpy.zeros((N, 3), numpy.int) a[:, 1] = 1 a[:, 2] = 2 cells = array_handler.array2vtkCellArray(a) arr = array_handler.vtk2array(cells.GetData()) expect = numpy.array([3, 0, 1, 2] * 3, numpy.int) self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)), True) self.assertEqual(cells.GetNumberOfCells(), N) # Test if a list of Numeric arrays of different cell lengths works. l_a = [a[:, :1], a, a[:2, :2]] cells = array_handler.array2vtkCellArray(l_a) arr = array_handler.vtk2array(cells.GetData()) expect = numpy.array([1, 0] * 3 + [3, 0, 1, 2] * 3 + [2, 0, 1] * 2, numpy.int) self.assertEqual(numpy.alltrue(numpy.equal(arr, expect)), True) self.assertEqual(cells.GetNumberOfCells(), N * 2 + 2) # This should not take a long while. This merely tests if a # million cells can be created rapidly. N = int(1e6) a = numpy.zeros((N, 3), numpy.int) a[:, 1] = 1 a[:, 2] = 2 cells = array_handler.array2vtkCellArray(a) self.assertEqual(cells.GetNumberOfCells(), N)
def test_deref_array(self): """Test if dereferencing array args works correctly.""" sigs = [[['vtkDataArray']], [['vtkFloatArray']], [['vtkCellArray']], [['vtkPoints']], [['int', 'vtkIdList']], [['int', ('float', 'float'), 'vtkDataArray']], [['Prop', 'int', 'vtkDataArray']], [['Points', ('float', 'float', 'float')]] ] args = [[[1,2,3]], [[0,0,0]], [[[1,2,3],[4,5,6]]], [[[0.,0.,0.], [1.,1.,1.]]], [1, [1,2,3]], [1, (0.0, 0.0), [1.0, 1.0, 1.0]], [Prop(), 1, numpy.array([1.0, 1.0, 1.0])], [[[1,2,3]], [1,2,3]] ] r = array_handler.deref_array(args[0], sigs[0]) self.assertEqual(mysum(array_handler.vtk2array(r[0]) -args[0]), 0) r = array_handler.deref_array(args[1], sigs[1]) self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[1]), 0) r = array_handler.deref_array(args[2], sigs[2]) self.assertEqual(r[0].GetNumberOfCells(), 2) r = array_handler.deref_array(args[3], sigs[3]) self.assertEqual(mysum(array_handler.vtk2array(r[0].GetData()) - numpy.array(args[3], 'f')), 0) r = array_handler.deref_array(args[4], sigs[4]) self.assertEqual(r[0], 1) self.assertEqual(r[1].__class__.__name__, 'vtkIdList') r = array_handler.deref_array(args[5], sigs[5]) self.assertEqual(r[0], 1) self.assertEqual(r[1], (0.0, 0.0)) self.assertEqual(mysum(array_handler.vtk2array(r[2]) -args[5][2]), 0) r = array_handler.deref_array(args[6], sigs[6]) self.assertEqual(r[0].IsA('vtkProperty'), True) self.assertEqual(r[1], 1) self.assertEqual(mysum(array_handler.vtk2array(r[2]) -args[6][2]), 0) r = array_handler.deref_array(args[7], sigs[7])
def test_id_array(self): """Test if a vtkIdTypeArray is converted correctly.""" arr = vtk.vtkIdTypeArray() arr.SetNumberOfTuples(10) for i in range(10): arr.SetValue(i, i) np = array_handler.vtk2array(arr) self.assertEqual(numpy.all(np == range(10)), True)
def test_reference_to_array(self): """Does to_array return an existing array instead of a new copy.""" arr = numpy.arange(0.0, 10.0, 0.1) arr = numpy.reshape(arr, (25, 4)) vtk_arr = array_handler.array2vtk(arr) arr1 = array_handler.vtk2array(vtk_arr) # Now make sure these are using the same memory. arr[0][0] = 100.0 self.assertEqual(arr[0][0], arr1[0][0]) self.assertEqual(arr.shape, arr1.shape)
def test_deref_array(self): """Test if dereferencing array args works correctly.""" sigs = [[['vtkDataArray']], [['vtkFloatArray']], [['vtkCellArray']], [['vtkPoints']], [['int', 'vtkIdList']], [['int', ('float', 'float'), 'vtkDataArray']], [['Prop', 'int', 'vtkDataArray']], [['Points', ('float', 'float', 'float')]]] args = [[[1, 2, 3]], [[0, 0, 0]], [[[1, 2, 3], [4, 5, 6]]], [[[0., 0., 0.], [1., 1., 1.]]], [1, [1, 2, 3]], [1, (0.0, 0.0), [1.0, 1.0, 1.0]], [Prop(), 1, numpy.array([1.0, 1.0, 1.0])], [[[1, 2, 3]], [1, 2, 3]]] r = array_handler.deref_array(args[0], sigs[0]) self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[0]), 0) r = array_handler.deref_array(args[1], sigs[1]) self.assertEqual(mysum(array_handler.vtk2array(r[0]) - args[1]), 0) r = array_handler.deref_array(args[2], sigs[2]) self.assertEqual(r[0].GetNumberOfCells(), 2) r = array_handler.deref_array(args[3], sigs[3]) self.assertEqual( mysum( array_handler.vtk2array(r[0].GetData()) - numpy.array(args[3], 'f')), 0) r = array_handler.deref_array(args[4], sigs[4]) self.assertEqual(r[0], 1) self.assertEqual(r[1].__class__.__name__, 'vtkIdList') r = array_handler.deref_array(args[5], sigs[5]) self.assertEqual(r[0], 1) self.assertEqual(r[1], (0.0, 0.0)) self.assertEqual(mysum(array_handler.vtk2array(r[2]) - args[5][2]), 0) r = array_handler.deref_array(args[6], sigs[6]) self.assertEqual(r[0].IsA('vtkProperty'), True) self.assertEqual(r[1], 1) self.assertEqual(mysum(array_handler.vtk2array(r[2]) - args[6][2]), 0) r = array_handler.deref_array(args[7], sigs[7])
def test_array2vtk(self): """Test Numeric array to VTK array conversion and vice-versa.""" # Put all the test arrays here. t_z = [] # Test the different types of arrays. t_z.append(numpy.array([-128, 0, 127], numpy.int8)) # FIXME: character arrays are a problem since there is no # unique mapping to a VTK data type and back. #t_z.append(numpy.array([-128, 0, 127], numpy.character)) t_z.append(numpy.array([-32768, 0, 32767], numpy.int16)) t_z.append(numpy.array([-2147483648, 0, 2147483647], numpy.int32)) t_z.append(numpy.array([0, 255], numpy.uint8)) t_z.append(numpy.array([0, 65535], numpy.uint16)) t_z.append(numpy.array([0, 4294967295L], numpy.uint32)) t_z.append(numpy.array([-1.0e38, 0, 1.0e38], 'f')) t_z.append(numpy.array([-1.0e299, 0, 1.0e299], 'd')) # Check multi-component arrays. t_z.append(numpy.array([[1], [2], [300]], 'd')) t_z.append(numpy.array([[1, 20], [300, 4000]], 'd')) t_z.append(numpy.array([[1, 2, 3], [4, 5, 6]], 'f')) t_z.append(numpy.array([[1, 2, 3],[4, 5, 6]], 'd')) t_z.append(numpy.array([[1, 2, 3, 400],[4, 5, 6, 700]], 'd')) t_z.append(numpy.array([range(9),range(10,19)], 'f')) # Test if a Python list also works. t_z.append(numpy.array([[1., 2., 3., 400.],[4, 5, 6, 700]], 'd')) # Test if arrays with number of components not in [1,2,3,4,9] work. t_z.append(numpy.array([[1, 2, 3, 400, 5000], [4, 5, 6, 700, 8000]], 'd')) t_z.append(numpy.array([range(10), range(10,20)], 'd')) for z in t_z: vtk_arr = array_handler.array2vtk(z) # Test for memory leaks. self.assertEqual(vtk_arr.GetReferenceCount(), array_handler.BASE_REFERENCE_COUNT) self._check_arrays(z, vtk_arr) z1 = array_handler.vtk2array(vtk_arr) if len(z.shape) == 1: self.assertEqual(len(z1.shape), 1) if z.dtype.char != 'c': #print z1 self.assertEqual(sum(numpy.ravel(z) - numpy.ravel(z1)), 0) else: #print z1.astype('c') self.assertEqual(z, z1.astype('c')) # Check if type conversion works correctly. z = numpy.array([-128, 0, 127], numpy.int8) vtk_arr = vtk.vtkDoubleArray() ident = id(vtk_arr) vtk_arr = array_handler.array2vtk(z, vtk_arr) # Make sure this is the same array! self.assertEqual(ident, id(vtk_arr)) self._check_arrays(z, vtk_arr) # Check the vtkBitArray. vtk_arr = vtk.vtkBitArray() vtk_arr.InsertNextValue(0) vtk_arr.InsertNextValue(1) vtk_arr.InsertNextValue(0) vtk_arr.InsertNextValue(1) arr = array_handler.vtk2array(vtk_arr) self.assertEqual(numpy.sum(arr - [0,1,0,1]), 0) vtk_arr = array_handler.array2vtk(arr, vtk_arr) self.assertEqual(vtk_arr.GetValue(0), 0) self.assertEqual(vtk_arr.GetValue(1), 1) self.assertEqual(vtk_arr.GetValue(2), 0) self.assertEqual(vtk_arr.GetValue(3), 1) # ---------------------------------------- # Test if the array is copied or not. a = numpy.array([[1, 2, 3],[4, 5, 6]], 'd') vtk_arr = array_handler.array2vtk(a) # Change the numpy array and see if the changes are # reflected in the VTK array. a[0] = [10.0, 20.0, 30.0] self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.)) # Make sure the cache is doing its job. key = vtk_arr.__this__ z = array_handler._array_cache.get(vtk_arr) self.assertEqual(numpy.sum(z - numpy.ravel(a)), 0.0) l1 = len(array_handler._array_cache) # del the Numeric array and see if this still works. del a self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.)) # Check the cache -- just making sure. self.assertEqual(len(array_handler._array_cache), l1) # Delete the VTK array and see if the cache is cleared. del vtk_arr self.assertEqual(len(array_handler._array_cache), l1-1) self.assertEqual(array_handler._array_cache._cache.has_key(key), False) # Make sure bit arrays are copied. vtk_arr = vtk.vtkBitArray() a = numpy.array([0,1,0,1], numpy.int32) vtk_arr = array_handler.array2vtk(a, vtk_arr) del a self.assertEqual(vtk_arr.GetValue(0), 0) self.assertEqual(vtk_arr.GetValue(1), 1) self.assertEqual(vtk_arr.GetValue(2), 0) self.assertEqual(vtk_arr.GetValue(3), 1) # Make sure the code at least runs for all the non-complex # numerical dtypes in numpy. for dtype in (numpy.sctypes['int'] + numpy.sctypes['uint'] + numpy.sctypes['float']): array_handler.array2vtk(numpy.zeros((1,), dtype=dtype))
def test_array2vtk(self): """Test Numeric array to VTK array conversion and vice-versa.""" # Put all the test arrays here. t_z = [] # Test the different types of arrays. t_z.append(numpy.array([-128, 0, 127], numpy.int8)) # FIXME: character arrays are a problem since there is no # unique mapping to a VTK data type and back. #t_z.append(numpy.array([-128, 0, 127], numpy.character)) t_z.append(numpy.array([-32768, 0, 32767], numpy.int16)) t_z.append(numpy.array([-2147483648, 0, 2147483647], numpy.int32)) t_z.append(numpy.array([0, 255], numpy.uint8)) t_z.append(numpy.array([0, 65535], numpy.uint16)) t_z.append(numpy.array([0, 4294967295L], numpy.uint32)) t_z.append(numpy.array([-1.0e38, 0, 1.0e38], 'f')) t_z.append(numpy.array([-1.0e299, 0, 1.0e299], 'd')) # Check multi-component arrays. t_z.append(numpy.array([[1], [2], [300]], 'd')) t_z.append(numpy.array([[1, 20], [300, 4000]], 'd')) t_z.append(numpy.array([[1, 2, 3], [4, 5, 6]], 'f')) t_z.append(numpy.array([[1, 2, 3], [4, 5, 6]], 'd')) t_z.append(numpy.array([[1, 2, 3, 400], [4, 5, 6, 700]], 'd')) t_z.append(numpy.array([range(9), range(10, 19)], 'f')) # Test if a Python list also works. t_z.append(numpy.array([[1., 2., 3., 400.], [4, 5, 6, 700]], 'd')) # Test if arrays with number of components not in [1,2,3,4,9] work. t_z.append( numpy.array([[1, 2, 3, 400, 5000], [4, 5, 6, 700, 8000]], 'd')) t_z.append(numpy.array([range(10), range(10, 20)], 'd')) for z in t_z: vtk_arr = array_handler.array2vtk(z) # Test for memory leaks. self.assertEqual(vtk_arr.GetReferenceCount(), array_handler.BASE_REFERENCE_COUNT) self._check_arrays(z, vtk_arr) z1 = array_handler.vtk2array(vtk_arr) if len(z.shape) == 1: self.assertEqual(len(z1.shape), 1) if z.dtype.char != 'c': #print z1 self.assertEqual(sum(numpy.ravel(z) - numpy.ravel(z1)), 0) else: #print z1.astype('c') self.assertEqual(z, z1.astype('c')) # Check if type conversion works correctly. z = numpy.array([-128, 0, 127], numpy.int8) vtk_arr = vtk.vtkDoubleArray() ident = id(vtk_arr) vtk_arr = array_handler.array2vtk(z, vtk_arr) # Make sure this is the same array! self.assertEqual(ident, id(vtk_arr)) self._check_arrays(z, vtk_arr) # Check the vtkBitArray. vtk_arr = vtk.vtkBitArray() vtk_arr.InsertNextValue(0) vtk_arr.InsertNextValue(1) vtk_arr.InsertNextValue(0) vtk_arr.InsertNextValue(1) arr = array_handler.vtk2array(vtk_arr) self.assertEqual(numpy.sum(arr - [0, 1, 0, 1]), 0) vtk_arr = array_handler.array2vtk(arr, vtk_arr) self.assertEqual(vtk_arr.GetValue(0), 0) self.assertEqual(vtk_arr.GetValue(1), 1) self.assertEqual(vtk_arr.GetValue(2), 0) self.assertEqual(vtk_arr.GetValue(3), 1) # ---------------------------------------- # Test if the array is copied or not. a = numpy.array([[1, 2, 3], [4, 5, 6]], 'd') vtk_arr = array_handler.array2vtk(a) # Change the numpy array and see if the changes are # reflected in the VTK array. a[0] = [10.0, 20.0, 30.0] self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.)) # Make sure the cache is doing its job. key = vtk_arr.__this__ z = array_handler._array_cache.get(vtk_arr) self.assertEqual(numpy.sum(z - numpy.ravel(a)), 0.0) l1 = len(array_handler._array_cache) # del the Numeric array and see if this still works. del a self.assertEqual(vtk_arr.GetTuple3(0), (10., 20., 30.)) # Check the cache -- just making sure. self.assertEqual(len(array_handler._array_cache), l1) # Delete the VTK array and see if the cache is cleared. del vtk_arr self.assertEqual(len(array_handler._array_cache), l1 - 1) self.assertEqual(array_handler._array_cache._cache.has_key(key), False) # Make sure bit arrays are copied. vtk_arr = vtk.vtkBitArray() a = numpy.array([0, 1, 0, 1], numpy.int32) vtk_arr = array_handler.array2vtk(a, vtk_arr) del a self.assertEqual(vtk_arr.GetValue(0), 0) self.assertEqual(vtk_arr.GetValue(1), 1) self.assertEqual(vtk_arr.GetValue(2), 0) self.assertEqual(vtk_arr.GetValue(3), 1) # Make sure the code at least runs for all the non-complex # numerical dtypes in numpy. for dtype in (numpy.sctypes['int'] + numpy.sctypes['uint'] + numpy.sctypes['float']): array_handler.array2vtk(numpy.zeros((1, ), dtype=dtype))