def test_arg_from_keywords():
kw_name = "framework_version"
kw_value = "1.6.0"
call = ast_call("MXNet({}='{}', py_version='py3', entry_point='run')".format(kw_name, kw_value))
returned_kw = parsing.arg_from_keywords(call, kw_name)
assert kw_name == returned_kw.arg
assert kw_value == returned_kw.value.s
def modify_node(self, node):
"""Modifies the ``ast.Call`` node to rename the attribute.
Args:
node (ast.Call): a node that represents the relevant function call.
Returns:
ast.AST: the original node, which has been potentially modified.
"""
keyword = parsing.arg_from_keywords(node, self.old_param_name)
keyword.arg = self.new_param_name
return node
def test_arg_from_keywords_absent_keyword():
call = ast_call("MXNet(entry_point='run')")
assert parsing.arg_from_keywords(call, "framework_version") is None