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
|
module Language where
import qualified Ast as A
import Compiler
import Lexer (scan)
import Parser (parse)
import Test.HUnit
import Text.Parsec (runParser)
import Text.Parsec.Pos (newPos)
assertAst :: String -> String -> [A.Expr] -> Assertion
assertAst tcase input expected = do
r <- return $ runParser (scan parse) () "test" input
case r of
Left e -> assertFailure $ show e
Right ast -> assertEqual tcase ast expected
testCase1 = TestCase $ assertAst "parse module" "module main" [A.Module "main" $ newPos "test" 1 1]
testCase2 =
TestCase $
assertAst
"parse a function"
"module main\n\
\def fn() { }"
[ A.Module "main" $ newPos "test" 1 1,
A.Func "fn" [] Nothing [] False False $ newPos "test" 2 1
]
testCase3 =
TestCase $
assertAst
"parse a function with parameters"
"module main\n\
\def fn(a: u8) { }"
[ A.Module "main" $ newPos "test" 1 1,
A.Func "fn" [("a", A.Type "u8", newPos "test" 2 8)] Nothing [] False False $ newPos "test" 2 1
]
testCase4 =
TestCase $
assertAst
"parse a function with return value"
"module main\n\
\def fn(): u8 {\n\
\return 1; }"
[ A.Module "main" $ newPos "test" 1 1,
A.Func "fn" [] (Just $ A.Type "u8") [A.Return (Just $ A.Num 1 $ newPos "test" 3 8) $ newPos "test" 3 1] False False $ newPos "test" 2 1
]
testCase5 =
TestCase $
assertAst
"parse a function call"
"module main\n\
\def fn() { }\n\
\fn();"
[ A.Module "main" $ newPos "test" 1 1,
A.Func "fn" [] Nothing [] False False $ newPos "test" 2 1,
A.Call (A.Var "fn" $ newPos "test" 3 1) [] $ newPos "test" 3 1
]
testCase6 =
TestCase $
assertAst
"parse a function call with arguments"
"module main\n\
\def fn(a: u8) { }\n\
\fn(10);"
[ A.Module "main" $ newPos "test" 1 1,
A.Func "fn" [("a", A.Type "u8", newPos "test" 2 8)] Nothing [] False False $ newPos "test" 2 1,
A.Call (A.Var "fn" $ newPos "test" 3 1) [A.Num 10 $ newPos "test" 3 4] $ newPos "test" 3 1
]
testCase7 =
TestCase $
assertAst
"parse empty return on a function"
"module main\n\
\def fn() {\n\
\return; }"
[ A.Module "main" $ newPos "test" 1 1,
A.Func "fn" [] Nothing [A.Return Nothing $ newPos "test" 3 1] False False $ newPos "test" 2 1
]
language = [testCase1, testCase2, testCase3, testCase4, testCase5, testCase6, testCase7]
|
