Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-21 12:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ** associativity

Browse Source
This commit is contained in:
blueloveTH 2024-11-12 19:06:17 +08:00
parent cbae815d37
commit 6266161118
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/compiler/compiler.c
View File

@ -1652,7 +1652,12 @@ static Error* exprBinaryOp(Compiler* self) {
Error* err; Error* err;
int line = prev()->line; int line = prev()->line;
TokenIndex op = prev()->type; TokenIndex op = prev()->type;
check(parse_expression(self, rules[op].precedence + 1, false)); int precedence = rules[op].precedence;
if(op != TK_POW) {
// if not right associative, increase precedence
precedence += 1;
}
check(parse_expression(self, precedence, false));
BinaryExpr* e = BinaryExpr__new(line, op, false); BinaryExpr* e = BinaryExpr__new(line, op, false);
if(op == TK_IN || op == TK_NOT_IN) { if(op == TK_IN || op == TK_NOT_IN) {
e->lhs = Ctx__s_popx(ctx()); e->lhs = Ctx__s_popx(ctx());

4
tests/99_extras.py
View File

@ -47,3 +47,7 @@ assert A()[::, :] == (slice(None, None, None), slice(None, None, None))
assert A()[::, :2] == (slice(None, None, None), slice(None, 2, None)) assert A()[::, :2] == (slice(None, None, None), slice(None, 2, None))
assert A()['b':'c':1, :] == (slice('b', 'c', 1), slice(None, None, None)) assert A()['b':'c':1, :] == (slice('b', 'c', 1), slice(None, None, None))
assert A()[1:2, :A()[3:4, ::-1]] == (slice(1, 2, None), slice(None, (slice(3, 4, None), slice(None, None, -1)), None)) assert A()[1:2, :A()[3:4, ::-1]] == (slice(1, 2, None), slice(None, (slice(3, 4, None), slice(None, None, -1)), None))
# test right associative
assert 2**2**3 == 256
assert (2**2**3)**2 == 65536