1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
package com.seaofnodes.simple;
import com.seaofnodes.simple.codegen.CodeGen;
import org.junit.Test;
import static org.junit.Assert.*;
public class CompoundAssignmentTest {
private static final String[] OPS = {"&=", "|=", "^=", "<<=", ">>=", ">>>="};
private static long apply(String op, long lhs, long rhs) {
return switch (op) {
case "&=" -> lhs & rhs;
case "|=" -> lhs | rhs;
case "^=" -> lhs ^ rhs;
case "<<=" -> lhs << rhs;
case ">>=" -> lhs >> rhs;
case ">>>=" -> lhs >>> rhs;
default -> throw new AssertionError(op);
};
}
@Test public void testLocals() {
for (String op : OPS)
for (long arg : new long[]{-8, 0, 5, 64}) {
long expected = apply(op, arg, 3);
check("int x=arg; int r=(x " + op + " 3); return x*10+r;",
arg, expected*11);
}
}
@Test public void testRhsOnce() {
for (String op : OPS) {
long expected = apply(op, 5, 1);
check("int x=arg; int count=0; int r=(x " + op +
" (count=count+1)); return count*1000+x*10+r;", 5, 1000+expected*11);
}
}
@Test public void testNarrowingAndChaining() {
check("u8 x=255; x <<= 1; return x;", 0, 254);
check("u8 x=255; x ^= 256; return x;", 0, 255);
check("int x=1; int y=3; x |= y <<= 2; return x*100+y;", 0, 1312);
check("int x=-8; x >>= 65; return x;", 0, -4);
check("int x=-8; x >>>= 65; return x;", 0, Long.MAX_VALUE-3);
}
@Test public void testExistingOperators() {
check("int x=8; x+=2; x-=1; x*=3; x/=9; return x;", 0, 3);
check("int x=arg; int y=x++; int z=x--; return x*100+y*10+z;", 5, 556);
check("return (arg << 2) + (arg >> 1) + (arg >>> 1);", 8, 40);
check("return arg <= 8;", 8, 1);
check("return arg >= 8;", 8, 1);
}
@Test public void testRejectedAssignments() {
for (String op : OPS) {
rejected("val x=3; x " + op + " 1; return x;");
rejected("int x=3; x " + op + " 1.5; return x;");
}
}
private static void rejected(String src) {
try {
check(src, 0, 0);
fail("Accepted invalid assignment: " + src);
} catch (RuntimeException e) {
assertTrue("Must report a language error: " + e, e instanceof Parser.ParseException);
assertNotNull(e.getMessage());
assertFalse("Must report a language error: " + e, e.getMessage().contains("Not yet implemented"));
}
}
@Test public void testFields() {
for (String op : OPS) {
long expected = apply(op, -8, 3);
check("struct S { int x; }; S !s=new S; s.x=arg; int r=(s.x " +
op + " 3); return s.x*10+r;", -8, expected*11);
}
}
@Test public void testArrayAddressAndRhsOnce() {
for (String op : OPS) {
long expected = apply(op, 5, 1);
check("int[] !a=new int[2]; a[0]=arg; int i=0; int count=0; " +
"int r=(a[i++] " + op + " (count=count+1)); " +
"return i*1000+count*100+a[0]*10+r;", 5, 1100+expected*11);
}
}
private static void check(String src, long arg, long expected) {
var code = new CodeGen(src).parse().opto().typeCheck();
assertEquals(src, Long.toString(expected), Eval2.eval(code, arg));
}
}
|
