def test_get_nodes_with_not_exist_name(self, load_graph_record, name,
node_type):
"""Test getting nodes with not exist name."""
with pytest.raises(ParamValueError) as exc_info:
graph_processor = GraphProcessor(self._train_id,
self._mock_data_manager)
graph_processor.get_nodes(name, node_type)
if name:
assert "The node name is not in graph." in exc_info.value.message
else:
assert f'The node name "{name}" not in graph, node type is {node_type}.' in exc_info.value.message
def test_get_nodes_success(self, load_graph_record, name, node_type,
result_file):
"""Test getting nodes successfully."""
graph_processor = GraphProcessor(self._train_id,
self._mock_data_manager)
results = graph_processor.get_nodes(name, node_type)
self.compare_result_with_file(results, result_file)
def graph_nodes():
"""
Interface to get graph nodes.
Returns:
Response, which contains a JSON object.
"""
name = request.args.get('name', default=None)
node_type = request.args.get('type', default='name_scope')
tag = request.args.get("tag", default=None)
train_id = get_train_id(request)
graph_process = GraphProcessor(train_id, DATA_MANAGER, tag)
response = graph_process.get_nodes(name=name, node_type=node_type)
return jsonify(response)