From e7f71a3ebafac3bfd99f67f3b79a205afe3d96ae Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sun, 8 Jul 2018 20:30:31 +0200 Subject: [PATCH] Use one simple functions to check for funcdefs in diff parser --- parso/python/diff.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/parso/python/diff.py b/parso/python/diff.py index b841d28..ee33669 100644 --- a/parso/python/diff.py +++ b/parso/python/diff.py @@ -580,9 +580,7 @@ class _NodesStack(object): self._tos = _NodesStackNode(tree_node, self._tos) self._tos.add(list(tree_node.children)) self._update_tos(tree_node.children[-1]) - elif tree_node.type in ('decorated', 'classdef', 'funcdef', - 'async_funcdef') \ - or tree_node.type == 'async_stmt' and tree_node.children[0].type == 'funcdef': + elif _func_or_class_has_suite(tree_node): self._update_tos(tree_node.children[-1]) def close(self):