def test_non_balanced_stm(self):
inputs = [
"while (1) {",
"while (1) {} else",
"while (1) {} else {",
"if (1 {}",
"if 1 {}",
"x = 23",
"x = ",
]
for input in inputs:
with self.subTest(i=input):
with self.assertRaises(Exception):
parse_stm(input)
def test_parse_stm_pointer_assign(self):
s3e = "*n = 1;"
self.assertEqual(
parse_stm(s3e),
[StmExpr(ArithAssign(lvalue_pointer("n"), ArithLit(1)))])
s3f = "*n += 1;"
self.assertEqual(parse_stm(s3f), [
StmExpr(
ArithAssign(
lvalue_pointer("n"),
ArithBinop(ArithOp.Add,
ArithUnaryop(ArithUnaryOp.Deref, Var("n")),
ArithLit(1))))
])
def test_parse_stm_deref(self):
s3g = "n = &m;"
self.assertEqual(parse_stm(s3g), [
StmExpr(
ArithAssign(lvalue_var("n"),
ArithUnaryop(ArithUnaryOp.Addr, Var("m"))))
])
def test_nested_block(self):
body = """
long x;
if (0 == 0) {
long x = 1;
}
"""
stms = parse_stm(body)
local_vars = get_local_vars(stms)
self.assertEqual(local_vars, ['x', 'x'])
def test_renamer(self):
body1 = """
long x;
while (x == 0) {
int x;
if (x == 0) {
long x;
}
}
"""
body2 = """
long x;
while (x == 0) {
int x2;
if (x2 == 0) {
long x3;
}
}
"""
stms1 = parse_stm(body1)
stms2 = parse_stm(body2)
self.assertEqual(rename_vars(stms1), stms2)
def test_function(self):
body = """
long x;
long y = 1;
int m = 0;
if (x == 0) {
long z;
} else {
long t;
}
while (x == x) {
long a = 1;
}
"""
stms = parse_stm(body)
local_vars = get_local_vars(stms)
self.assertEqual(local_vars, ['x', 'y', 'm', 'z', 't', 'a'])
def test_parse_stm_neg(self):
s1 = "print(!x);"
self.assertEqual(parse_stm(s1), [StmPrint(BoolNeg(Var("x")))])
def test_parser_stm(self):
s1 = "print(x);"
self.assertEqual(parse_stm(s1), [StmPrint(Var("x"))])
s1a = "print(x_);"
self.assertEqual(parse_stm(s1a), [StmPrint(Var("x_"))])
s2 = "print(15);"
self.assertEqual(parse_stm(s2), [StmPrint(ArithLit(15))])
s3 = "x = 1;"
self.assertEqual(parse_stm(s3),
[StmExpr(ArithAssign(lvalue_var("x"), ArithLit(1)))])
s3a = "x += 1;"
self.assertEqual(parse_stm(s3a), [
StmExpr(
ArithAssign(lvalue_var("x"),
ArithBinop(ArithOp.Add, Var("x"), ArithLit(1))))
])
s3b = "x -= 1;"
self.assertEqual(parse_stm(s3b), [
StmExpr(
ArithAssign(lvalue_var("x"),
ArithBinop(ArithOp.Sub, Var("x"), ArithLit(1))))
])
s3c = "x *= 1;"
self.assertEqual(parse_stm(s3c), [
StmExpr(
ArithAssign(lvalue_var("x"),
ArithBinop(ArithOp.Mul, Var("x"), ArithLit(1))))
])
s3d = "x /= 1;"
self.assertEqual(parse_stm(s3d), [
StmExpr(
ArithAssign(lvalue_var("x"),
ArithBinop(ArithOp.Div, Var("x"), ArithLit(1))))
])
s3d = "x %= 1;"
self.assertEqual(parse_stm(s3d), [
StmExpr(
ArithAssign(lvalue_var("x"),
ArithBinop(ArithOp.Mod, Var("x"), ArithLit(1))))
])
s4 = "while (x == 1) {}"
self.assertEqual(
parse_stm(s4),
[StmWhile(BoolArithCmp(ArithCmp.Eq, Var("x"), ArithLit(1)), [])])
s5 = "while (x == 1) { x = x + 1; }"
self.assertEqual(parse_stm(s5), [
StmWhile(BoolArithCmp(ArithCmp.Eq, Var("x"), ArithLit(1)), [
StmExpr(
ArithAssign(lvalue_var("x"),
ArithBinop(ArithOp.Add, Var("x"),
ArithLit(1))))
])
])
s6 = "if (x == 1) {}"
self.assertEqual(
parse_stm(s6),
[StmIf(BoolArithCmp(ArithCmp.Eq, Var("x"), ArithLit(1)), [], [])])
s7 = "if (x == 1) {} else {}"
self.assertEqual(
parse_stm(s7),
[StmIf(BoolArithCmp(ArithCmp.Eq, Var("x"), ArithLit(1)), [], [])])
s8 = "{ }"
self.assertEqual(parse_stm(s8), [StmBlock([])])
s9 = "break;"
self.assertEqual(parse_stm(s9), [StmBreak()])
s10 = "continue;"
self.assertEqual(parse_stm(s10), [StmContinue()])
def test_parser_decl_initialized(self):
decl = "long x = 123;"
self.assertEqual(
parse_stm(decl),
[StmDecl(tp_normal(AtomType.Long), 'x', ArithLit(num=123))])
def test_parser_decl_pointer_initialized_address(self):
decl = "long* x = &n;"
self.assertEqual(parse_stm(decl), [
StmDecl(tp_pointer(AtomType.Long), "x",
ArithUnaryop(ArithUnaryOp.Addr, Var("n")), VarKind.Local)
])
def test_parser_decl_pointer_initialized(self):
decl = "long* x = 0;"
self.assertEqual(parse_stm(decl), [
StmDecl(tp_pointer(AtomType.Long), "x", ArithLit(0), VarKind.Local)
])
def test_parser_decl_pointer(self):
decl = "long* x;"
self.assertEqual(
parse_stm(decl),
[StmDecl(tp_pointer(AtomType.Long), "x", None, VarKind.Local)])
def test_parser_decl(self):
decl = "long x;"
self.assertEqual(
parse_stm(decl),
[StmDecl(type=tp_normal(AtomType.Long), var='x', a=None)])