Method test

Serialization/Manager/Testing/Python/TestBlobs.py:6–83

(self)

class TestBlobs(vtkTesting.vtkTest):

    def test(self):
        from vtkmodules.vtkSerializationManager import vtkObjectManager
        from vtkmodules.vtkCommonCore import vtkTypeFloat32Array, vtkUnsignedIntArray

        arrayF32Id = ""
        arrayUI32Id = ""

        def serialize():
            global arrayF32Id, arrayUI32Id

            manager = vtkObjectManager()
            if not manager.Initialize():
                return


            arrayF32 = vtkTypeFloat32Array()
            arrayUI32 = vtkUnsignedIntArray()
            for i in range(10):
                arrayF32.InsertNextValue(10 - i)
                arrayUI32.InsertNextValue(20 + i)

            arrayF32Id = manager.RegisterObject(arrayF32)
            self.assertEqual(arrayF32Id, 1)

            arrayUI32Id = manager.RegisterObject(arrayUI32)
            self.assertEqual(arrayUI32Id, 2)

            manager.UpdateStatesFromObjects()
            active_ids = manager.GetAllDependencies(0)

            # these are not guaranteed to appear in the exact same order
            # due to the unordered_map in use within vtkObjectManager.
            self.assertEqual(len(active_ids), 2)
            self.assertEqual(arrayF32Id in active_ids, True)
            self.assertEqual(arrayUI32Id in active_ids, True)

            states = [ manager.GetState(object_id) for object_id in active_ids ]
            hash_to_blob_map = { blob_hash: manager.GetBlob(blob_hash, True) for blob_hash in manager.GetBlobHashes(active_ids) }

            manager.UnRegisterObject(arrayF32Id)
            manager.UnRegisterObject(arrayUI32Id)

            return states, hash_to_blob_map

        def deserialize(states, hash_to_blob_map):
            global arrayF32Id, arrayUI32Id

            if not len(states):
                return

            manager = vtkObjectManager()
            if not manager.Initialize():
                return

            for state in states:
                manager.RegisterState(state)
            for hash_text, blob in hash_to_blob_map.items():
                manager.RegisterBlob(hash_text, blob)