def testExceptions(self):
"Test if the right assertion errors are raised."
# Test if bit arrays raise an exception.
svtk_arr = svtk.svtkBitArray()
svtk_arr.InsertNextValue(0)
svtk_arr.InsertNextValue(1)
self.assertRaises(AssertionError, svtk_to_numpy, svtk_arr)
def testBitArrayArguments(self):
a = svtk.svtkBitArray()
a.SetNumberOfComponents(2)
a.SetNumberOfTuples(1)
ti = [0, 1]
to = [0, 0]
a.SetTuple(0, ti)
a.GetTuple(0, to)
self.assertEqual(ti, [int(x) for x in to])
def testBitArray(self):
"""Test the special case of the bit array."""
if sys.hexversion < 0x02070000:
return
# bit array is actually stored as a byte array
a = svtk.svtkBitArray()
a.InsertNextValue(0)
a.InsertNextValue(1)
a.InsertNextValue(1)
a.InsertNextValue(0)
a.InsertNextValue(1)
m = memoryview(a)
self.assertEqual(m.format, 'B')
self.assertEqual(m.itemsize, 1)
self.assertEqual(m.shape, (1, ))
# test the contents of the memoryview
self.assertEqual(ord(m.tobytes()) & 0xF8, 0x68)
without mask
during HTG build we build scalar, used Global Index implicit with SetGlobalIndexStart
SetGlobalIndexStart, one call by HT
"""
import svtk
htg = svtk.svtkHyperTreeGrid()
htg.Initialize()
scalarArray = svtk.svtkDoubleArray()
scalarArray.SetName('scalar')
scalarArray.SetNumberOfValues(0)
htg.GetPointData().AddArray(scalarArray)
htg.GetPointData().SetActiveScalars('scalar')
mask = svtk.svtkBitArray()
mask.SetName('mask')
mask.SetNumberOfValues(52)
mask.FillComponent(0, 0)
htg.SetMask(mask)
htg.SetDimensions([4, 3, 3])
htg.SetBranchFactor(2)
# Rectilinear grid coordinates
xValues = svtk.svtkDoubleArray()
xValues.SetNumberOfValues(4)
xValues.SetValue(0, -1)
xValues.SetValue(1, 0)
xValues.SetValue(2, 1)
xValues.SetValue(3, 2)