def testParseConcatExpr(self):
self.assertEqual(
ParseExprFromStr('老王加油、2、“哈”')[0],
ConcatExpr([
ArithmeticExpr(VariableExpr('老王'), Keyword('加'),
VariableExpr('油')),
IntegerLiteralExpr(2),
StringLiteralExpr('哈')
]))
def testParseTermExpr(self):
self.assertEqual(
ParseExprFromStr(u'老王乘五')[0],
ArithmeticExpr(VariableExpr(Token(TK_IDENTIFIER, u'老王')),
Keyword(u'乘'),
LiteralExpr(Token(TK_INTEGER_LITERAL, 5))))
self.assertEqual(
ParseExprFromStr(u'五除以老王')[0],
ArithmeticExpr(LiteralExpr(Token(TK_INTEGER_LITERAL, 5)),
Keyword(u'除以'),
VariableExpr(Token(TK_IDENTIFIER, u'老王'))))
self.assertEqual(
ParseExprFromStr(u'五除以老王乘老刘')[0],
ArithmeticExpr(
ArithmeticExpr(LiteralExpr(Token(TK_INTEGER_LITERAL, 5)),
Keyword(u'除以'),
VariableExpr(Token(TK_IDENTIFIER, u'老王'))),
Keyword(u'乘'), VariableExpr(Token(TK_IDENTIFIER, u'老刘'))))
def testParseConcatExpr(self):
self.assertEqual(
ParseExprFromStr('老王加油、2、“哈”')[0],
ConcatExpr([
ArithmeticExpr(VariableExpr(Token(TK_IDENTIFIER, '老王')),
Keyword('加'),
VariableExpr(Token(TK_IDENTIFIER, '油'))),
LiteralExpr(Token(TK_INTEGER_LITERAL, 2)),
LiteralExpr(Token(TK_STRING_LITERAL, '哈'))
]))
def testParseArithmeticExpr(self):
self.assertEqual(
ParseExprFromStr('5加六')[0],
ArithmeticExpr(LiteralExpr(Token(TK_INTEGER_LITERAL, 5)),
Keyword('加'),
LiteralExpr(Token(TK_INTEGER_LITERAL, 6))))
self.assertEqual(
ParseExprFromStr('5加六乘3')[0],
ArithmeticExpr(
LiteralExpr(Token(TK_INTEGER_LITERAL, 5)), Keyword('加'),
ArithmeticExpr(LiteralExpr(Token(TK_INTEGER_LITERAL, 6)),
Keyword('乘'),
LiteralExpr(Token(TK_INTEGER_LITERAL, 3)))))
self.assertEqual(
ParseExprFromStr('5减六减老王')[0],
ArithmeticExpr(
ArithmeticExpr(LiteralExpr(Token(TK_INTEGER_LITERAL, 5)),
Keyword('减'),
LiteralExpr(Token(TK_INTEGER_LITERAL, 6))),
Keyword('减'), VariableExpr(Token(TK_IDENTIFIER, '老王'))))
def testParseComparisonExpr(self):
self.assertEqual(
ParseExprFromStr('5比6大')[0],
ComparisonExpr(IntegerLiteralExpr(5), Keyword('大'),
IntegerLiteralExpr(6)))
self.assertEqual(
ParseExprFromStr('老王加5比6小')[0],
ComparisonExpr(
ArithmeticExpr(VariableExpr('老王'), Keyword('加'),
IntegerLiteralExpr(5)), Keyword('小'),
IntegerLiteralExpr(6)))
self.assertEqual(
ParseExprFromStr('老王跟老刘一样一样的')[0],
ComparisonExpr(VariableExpr('老王'), Keyword('一样一样的'),
VariableExpr('老刘')))
self.assertEqual(
ParseExprFromStr('老王加5跟6不是一样一样的')[0],
ComparisonExpr(
ArithmeticExpr(VariableExpr('老王'), Keyword('加'),
IntegerLiteralExpr(5)), Keyword('不是一样一样的'),
IntegerLiteralExpr(6)))
def testParseTermExpr(self):
self.assertEqual(ParseExprFromStr('老王乘五')[0],
ArithmeticExpr(
VariableExpr('老王'),
Keyword('乘'),
IntegerLiteralExpr(5))
)
self.assertEqual(ParseExprFromStr('五除以老王')[0],
ArithmeticExpr(
IntegerLiteralExpr(5),
Keyword('除以'),
VariableExpr('老王'))
)
self.assertEqual(ParseExprFromStr('五除以老王乘老刘')[0],
ArithmeticExpr(
ArithmeticExpr(
IntegerLiteralExpr(5),
Keyword('除以'),
VariableExpr('老王')),
Keyword('乘'),
VariableExpr('老刘')
))
def testParsingLoop(self):
self.assertEqual(
ParseToAst('老王从1到9磨叽：磨叽完了。'),
[Statement(
STMT_LOOP,
(VariableExpr('老王'),
NumberLiteralExpr(1),
NumberLiteralExpr(9),
NumberLiteralExpr(1),
[]))])
self.assertEqual(
ParseToAst('老王从二到十一步七蹿磨叽：磨叽完了。'),
[Statement(
STMT_LOOP,
(VariableExpr('老王'),
NumberLiteralExpr(2),
NumberLiteralExpr(10),
NumberLiteralExpr(7),
[]))])
self.assertEqual(
ParseToAst('老王从二到十一步七减一蹿磨叽：磨叽完了。'),
[Statement(
STMT_LOOP,
(VariableExpr('老王'),
NumberLiteralExpr(2),
NumberLiteralExpr(10),
ArithmeticExpr(
NumberLiteralExpr(7),
Keyword('减'),
NumberLiteralExpr(1)),
[]))])
self.assertEqual(
ParseToAst('老王从一而终磨叽：磨叽完了。'),
[Statement(
STMT_INFINITE_LOOP,
(VariableExpr('老王'),
[]))])
self.assertEqual(
ParseToAst('老张在苹果总部磨叽：磨叽完了。'),
[Statement(
STMT_INFINITE_LOOP,
(VariableExpr('老张'),
[]))])