added new graph metrics
This commit is contained in:
@@ -121,6 +121,7 @@ def test_pipeline_valid(pipeline, alter_content):
|
||||
assert len(pipe.actions) == 1
|
||||
assert len(pipe.action_names) == 1
|
||||
assert len(pipe.actions_kwargs) == 1
|
||||
assert len(pipe.action_skip) == 1
|
||||
assert len(pipe.save_results) == 1
|
||||
assert len(pipe.load_results) == 1
|
||||
assert pipe.save_results[0] == (False, None)
|
||||
@@ -166,7 +167,7 @@ def test_pipeline_valid_action_load(pipeline, working_dir):
|
||||
test_string = 'test'
|
||||
|
||||
# action preparation
|
||||
def valid_action(string, add_content=False):
|
||||
def valid_action(string, add_content=False): # pragma: no cover
|
||||
if add_content:
|
||||
string += '_2'
|
||||
return string
|
||||
@@ -175,6 +176,7 @@ def test_pipeline_valid_action_load(pipeline, working_dir):
|
||||
assert len(pipe.actions) == 1
|
||||
assert len(pipe.action_names) == 1
|
||||
assert len(pipe.actions_kwargs) == 1
|
||||
assert len(pipe.action_skip) == 1
|
||||
assert len(pipe.save_results) == 1
|
||||
assert len(pipe.load_results) == 1
|
||||
assert pipe.save_results[0] == (False, None)
|
||||
@@ -209,19 +211,28 @@ def test_pipeline_multiple_actions(pipeline):
|
||||
string += '_3'
|
||||
return string
|
||||
|
||||
pipe.add(valid_action, {'add_content': True}, skip=True)
|
||||
pipe.add(valid_action, {'add_content': True})
|
||||
pipe.add(valid_action_2)
|
||||
assert len(pipe.actions) == 2
|
||||
assert len(pipe.action_names) == 2
|
||||
assert len(pipe.actions_kwargs) == 2
|
||||
assert len(pipe.save_results) == 2
|
||||
assert len(pipe.load_results) == 2
|
||||
assert len(pipe.actions) == 3
|
||||
assert len(pipe.action_names) == 3
|
||||
assert len(pipe.actions_kwargs) == 3
|
||||
assert len(pipe.action_skip) == 3
|
||||
assert len(pipe.save_results) == 3
|
||||
assert len(pipe.load_results) == 3
|
||||
assert pipe.save_results[1] == (False, None)
|
||||
assert pipe.load_results[1] == (False, None)
|
||||
|
||||
ret = pipe.run(starting_values=(test_string,))
|
||||
assert isinstance(ret, tuple)
|
||||
assert pipe._intermediate_result == ret
|
||||
assert pipe.curr_proc_idx == 3
|
||||
assert pipe.curr_proc_idx == 4
|
||||
assert ret is not None
|
||||
assert ret[0] == 'test_2_3'
|
||||
|
||||
|
||||
def test_pipeline_invalid_action(pipeline):
|
||||
test_string = 'test'
|
||||
|
||||
with pytest.raises(WrongActionTypeError):
|
||||
pipeline.add(test_string, skip=False)
|
||||
|
||||
Reference in New Issue
Block a user