diff --git a/ASTree.cpp b/ASTree.cpp index 7f36b3d..d66859b 100644 --- a/ASTree.cpp +++ b/ASTree.cpp @@ -431,8 +431,8 @@ PycRef BuildFromCode(PycRef code, PycModule* mod) bases.resize(basecnt); PycRef TOS = stack.top(); int TOS_type = TOS.type(); - // bases are NODE_NAME at TOS - while (TOS_type == ASTNode::NODE_NAME) { + // bases are NODE_NAME and NODE_BINARY at TOS + while (TOS_type == ASTNode::NODE_NAME || TOS_type == ASTNode::NODE_BINARY) { bases.resize(basecnt + 1); bases[basecnt] = TOS; basecnt++; diff --git a/tests/compiled/class_NODE_BINARY.3.9.pyc b/tests/compiled/class_NODE_BINARY.3.9.pyc new file mode 100644 index 0000000..1f2ccc1 Binary files /dev/null and b/tests/compiled/class_NODE_BINARY.3.9.pyc differ diff --git a/tests/input/class_NODE_BINARY.py b/tests/input/class_NODE_BINARY.py new file mode 100644 index 0000000..9d5c613 --- /dev/null +++ b/tests/input/class_NODE_BINARY.py @@ -0,0 +1,3 @@ +import Test +class MyNet(Test.t): + pass diff --git a/tests/tokenized/class_NODE_BINARY.txt b/tests/tokenized/class_NODE_BINARY.txt new file mode 100644 index 0000000..71b3977 --- /dev/null +++ b/tests/tokenized/class_NODE_BINARY.txt @@ -0,0 +1,4 @@ +import Test +class MyNet ( Test . t ) : + +pass