def test_queue_up(self): inputs = {'a': Int(2), 'b': Str('test')} # Queue up the process pid = queue_up(DummyProcess, inputs, self.storage) # Then load the checkpoint and instantiate the class cp = self.storage.load_checkpoint(pid) dp = DummyProcess.create_from(cp) self.assertIsInstance(dp, DummyProcess) self.assertEqual(dp.raw_inputs, inputs) dp.run_until_complete()
def test_seal(self): rinfo = run(DummyProcess, _return_pid=True)[1] self.assertTrue(load_node(pk=rinfo).is_sealed) storedir = tempfile.mkdtemp() storage = Persistence.create_from_basedir(storedir) rinfo = submit(DummyProcess, _jobs_store=storage) n = load_node(pk=rinfo.pid) self.assertFalse(n.is_sealed) dp = DummyProcess.create_from(storage.load_checkpoint(rinfo.pid)) dp.run_until_complete() self.assertTrue(n.is_sealed) shutil.rmtree(storedir)