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.0, 0.0], [1.0, 1.0, 1.0]]],
[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_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_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])