def test_to_json_dict(self): resource = get_resource('workflows/three_ops.json') workflow = Workflow.load(resource) step = WorkflowStep(workflow, resource, node_id='jojo_87') actual_json_dict = step.to_json_dict() expected_json_text = """ { "id": "jojo_87", "workflow": "%s", "inputs": { "p": {} }, "outputs": { "q": {} } } """ % resource actual_json_text = json.dumps(actual_json_dict, indent=4) expected_json_dict = json.loads(expected_json_text) actual_json_dict = json.loads(actual_json_text) self.assertEqual(actual_json_dict, expected_json_dict, msg='\n%sexpected:\n%s\n%s\nbut got:\n%s\n' % (120 * '-', expected_json_text, 120 * '-', actual_json_text))
def test_to_json_dict(self): resource = get_resource('workflows/three_ops.json') workflow = Workflow.load(resource) step = WorkflowStep(workflow, resource, node_id='jojo_87') actual_json_dict = step.to_json_dict() expected_json_text = """ { "id": "jojo_87", "workflow": "%s", "inputs": { "p": {} }, "outputs": { "q": {} } } """ % resource actual_json_text = json.dumps(actual_json_dict, indent=4) expected_json_dict = json.loads(expected_json_text) actual_json_dict = json.loads(actual_json_text) self.assertEqual( actual_json_dict, expected_json_dict, msg='\n%sexpected:\n%s\n%s\nbut got:\n%s\n' % (120 * '-', expected_json_text, 120 * '-', actual_json_text))
def test_invoke(self): resource = get_resource('workflows/three_ops.json') workflow = Workflow.load(resource) step = WorkflowStep(workflow, resource, node_id='jojo_87') value_cache = {} step.inputs.p.value = 3 step.invoke(context=dict(value_cache=value_cache)) output_value = step.outputs.q.value self.assertEqual(output_value, 2 * (3 + 1) + 3 * (2 * (3 + 1))) self.assertEqual(value_cache, {'op1': {'y': 4}, 'op2': {'b': 8}, 'op3': {'w': 32}})
def test_invoke(self): resource = get_resource('workflows/three_ops.json') workflow = Workflow.load(resource) step = WorkflowStep(workflow, resource, node_id='jojo_87') value_cache = {} step.inputs.p.value = 3 step.invoke(context=dict(value_cache=value_cache)) output_value = step.outputs.q.value self.assertEqual(output_value, 2 * (3 + 1) + 3 * (2 * (3 + 1))) self.assertEqual(value_cache, {'op1': {'y': 4}, 'op2': {'b': 8}, 'op3': {'w': 32}})
def test_invoke_as_part_of_workflow(self): resource = get_resource('workflows/three_ops.json') workflow = Workflow.load(resource) step = WorkflowStep(workflow, resource, node_id='jojo_87') workflow = Workflow( OpMetaInfo('contains_jojo_87', has_monitor=True, inputs=OrderedDict(x={}), outputs=OrderedDict(y={}))) workflow.add_step(step) step.inputs.p.source = workflow.inputs.x workflow.outputs.y.source = step.outputs.q value_cache = ValueCache() workflow.inputs.x.value = 4 workflow.invoke(context=dict(value_cache=value_cache)) output_value = workflow.outputs.y.value self.assertEqual(output_value, 2 * (4 + 1) + 3 * (2 * (4 + 1))) self.assertEqual( value_cache, { 'jojo_87._child': { 'op1': { 'y': 5 }, 'op2': { 'b': 10 }, 'op3': { 'w': 40 } } })
def test_from_json_dict(self): resource = get_resource('workflows/three_ops.json') json_text = """ { "id": "workflow_ref_89", "workflow": "%s", "inputs": { "p": {"value": 2.8} } } """ % resource json_dict = json.loads(json_text) step = WorkflowStep.from_json_dict(json_dict) self.assertIsInstance(step, WorkflowStep) self.assertEqual(step.id, "workflow_ref_89") self.assertEqual(step.resource, resource) self.assertIn('p', step.inputs) self.assertIn('q', step.outputs) self.assertEqual(step.inputs.p.value, 2.8) self.assertEqual(step.outputs.q.value, None) self.assertIsNotNone(step.workflow) self.assertIn('p', step.workflow.inputs) self.assertIn('q', step.workflow.outputs) self.assertIs(step.workflow.inputs.p.source, step.inputs.p)
def test_from_json_dict(self): resource = get_resource('workflows/three_ops.json') json_text = """ { "id": "workflow_ref_89", "workflow": "%s", "inputs": { "p": {"value": 2.8} } } """ % resource json_dict = json.loads(json_text) step = WorkflowStep.from_json_dict(json_dict) self.assertIsInstance(step, WorkflowStep) self.assertEqual(step.id, "workflow_ref_89") self.assertEqual(step.resource, resource) self.assertIn('p', step.inputs) self.assertIn('q', step.outputs) self.assertEqual(step.inputs.p.value, 2.8) self.assertEqual(step.outputs.q.value, None) self.assertIsNotNone(step.workflow) self.assertIn('p', step.workflow.inputs) self.assertIn('q', step.workflow.outputs) self.assertIs(step.workflow.inputs.p.source, step.inputs.p)
def test_init(self): resource = get_resource('workflows/three_ops.json') workflow = Workflow.load(resource) step = WorkflowStep(workflow, resource, node_id='jojo_87') self.assertEqual(step.id, 'jojo_87') self.assertEqual(step.resource, resource) self.assertEqual(str(step), 'jojo_87 = cool_workflow(p=None) -> (q) [WorkflowStep]') self.assertEqual(repr(step), "WorkflowStep(Workflow('cool_workflow'), '%s', node_id='jojo_87')" % resource) self.assertIsNotNone(step.workflow) self.assertIn('p', step.workflow.inputs) self.assertIn('q', step.workflow.outputs)