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