diff --git a/src/Evalvisitor.cpp b/src/Evalvisitor.cpp index d342d5f..757b8e8 100644 --- a/src/Evalvisitor.cpp +++ b/src/Evalvisitor.cpp @@ -202,7 +202,7 @@ std::any EvalVisitor::visitCompound_stmt( std::any EvalVisitor::visitIf_stmt(Python3Parser::If_stmtContext *ctx) { auto test_list = ctx->test(); auto suite_list = ctx->suite(); - if (test_list.size() - suite_list.size() != 1 && + if (test_list.size() - suite_list.size() != -1 && test_list.size() - suite_list.size() != 0) throw InterpretException("test and suite doesn't match in If_stmt"); for (int i = 0; i < test_list.size(); i++)