Mercurial > lcfOS
annotate test/testc3.py @ 306:b145f8e6050b
Start on c3 rewrite
| author | Windel Bouwman |
|---|---|
| date | Mon, 09 Dec 2013 19:00:21 +0100 |
| parents | be7f60545368 |
| children | e609d5296ee9 |
| rev | line source |
|---|---|
| 300 | 1 from ppci.c3 import Builder, Lexer |
| 301 | 2 from target import SimpleTarget |
| 287 | 3 import ppci |
| 167 | 4 import unittest |
| 287 | 5 import io |
| 148 | 6 |
| 7 | |
| 204 | 8 class testLexer(unittest.TestCase): |
| 293 | 9 def setUp(self): |
| 10 diag = ppci.DiagnosticsManager() | |
| 300 | 11 self.l = Lexer(diag) |
| 293 | 12 |
| 205 | 13 def testUnexpectedCharacter(self): |
| 287 | 14 snippet = io.StringIO(""" var s \u6c34 """) |
| 205 | 15 with self.assertRaises(ppci.CompilerError): |
| 293 | 16 list(self.l.tokenize(snippet)) |
| 17 | |
| 18 def check(self, snippet, toks): | |
| 19 toks2 = list(tok.typ for tok in self.l.tokenize(io.StringIO(snippet))) | |
| 20 self.assertSequenceEqual(toks, toks2) | |
| 205 | 21 |
| 204 | 22 def testBlockComment(self): |
| 293 | 23 snippet = """ |
| 204 | 24 /* Demo */ |
| 25 var int x = 0; | |
| 293 | 26 """ |
| 204 | 27 toks = ['var', 'ID', 'ID', '=', 'NUMBER', ';', 'END'] |
| 293 | 28 self.check(snippet, toks) |
| 205 | 29 |
| 204 | 30 def testBlockCommentMultiLine(self): |
| 293 | 31 snippet = """ |
| 204 | 32 /* Demo |
| 33 bla1 | |
| 34 bla2 | |
| 35 */ | |
| 36 var int x = 0; | |
| 293 | 37 """ |
| 204 | 38 toks = ['var', 'ID', 'ID', '=', 'NUMBER', ';', 'END'] |
| 293 | 39 self.check(snippet, toks) |
| 204 | 40 |
| 287 | 41 |
| 204 | 42 class testBuilder(unittest.TestCase): |
| 215 | 43 def setUp(self): |
| 44 self.diag = ppci.DiagnosticsManager() | |
| 301 | 45 self.builder = Builder(self.diag, SimpleTarget()) |
| 215 | 46 self.diag.clear() |
| 47 | |
| 306 | 48 def makeFileList(self, snippet): |
| 49 """ Try to make a list with opened files """ | |
| 50 if type(snippet) is list: | |
| 51 l2 = [] | |
| 52 for s in snippet: | |
| 53 if type(s) is str: | |
| 54 l2.append(io.StringIO(s)) | |
| 55 else: | |
| 56 l2.append(s) | |
| 57 return l2 | |
| 58 else: | |
| 59 return [io.StringIO(snippet)] | |
| 60 | |
| 215 | 61 def expectErrors(self, snippet, rows): |
| 62 """ Helper to test for expected errors on rows """ | |
| 287 | 63 ircode = list(self.builder.build([io.StringIO(snippet)])) |
| 215 | 64 actualErrors = [err.row for err in self.diag.diags] |
| 65 if rows != actualErrors: | |
| 306 | 66 self.diag.printErrors() |
| 215 | 67 self.assertSequenceEqual(rows, actualErrors) |
| 287 | 68 # self.assertFalse(all(ircode)) |
| 215 | 69 |
| 287 | 70 def expectOK(self, snippet): |
| 306 | 71 """ Expect a snippet to be OK """ |
| 72 ircode = list(self.builder.build(self.makeFileList(snippet))) | |
| 73 if len(self.diag.diags) > 0: | |
| 74 self.diag.printErrors() | |
| 287 | 75 self.assertTrue(all(ircode)) |
| 306 | 76 self.assertEqual(0, len(self.diag.diags)) |
| 217 | 77 return ircode |
| 215 | 78 |
|
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
249
diff
changeset
|
79 def testPackage(self): |
| 284 | 80 p1 = """module p1; |
|
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
249
diff
changeset
|
81 type int A; |
|
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
249
diff
changeset
|
82 """ |
| 284 | 83 p2 = """module p2; |
|
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
249
diff
changeset
|
84 import p1; |
| 306 | 85 var p1.A b; |
|
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
249
diff
changeset
|
86 """ |
| 306 | 87 self.expectOK([p1, p2]) |
|
251
6ed3d3a82a63
Added another c3 example. First import attempt
Windel Bouwman
parents:
249
diff
changeset
|
88 |
| 288 | 89 def testPackageMutual(self): |
| 90 p1 = """module p1; | |
| 91 import p2; | |
| 92 type int A; | |
| 93 var p2.B b; | |
| 94 """ | |
| 95 p2 = """module p2; | |
| 96 import p1; | |
| 97 var p1.A a; | |
| 98 """ | |
| 306 | 99 self.expectOK([p1, p2]) |
| 288 | 100 |
| 293 | 101 def testPackageNotExists(self): |
| 102 p1 = """module p1; | |
| 103 import p23; | |
| 104 """ | |
| 306 | 105 self.expectErrors(p1, [0]) |
| 293 | 106 |
| 215 | 107 def testFunctArgs(self): |
| 287 | 108 snippet = """ |
| 284 | 109 module testargs; |
| 167 | 110 function void t2(int a, double b) |
| 111 { | |
| 112 t2(2, 2); | |
| 113 t2(2); | |
| 114 t2(1, 1.2); | |
| 115 } | |
| 287 | 116 """ |
| 117 self.expectErrors(snippet, [5, 6]) | |
| 148 | 118 |
| 303 | 119 def testReturn(self): |
| 120 snippet = """ | |
| 121 module testreturn; | |
| 122 function void t() | |
| 123 { | |
| 124 return; | |
| 125 } | |
| 126 """ | |
| 127 self.expectOK(snippet) | |
| 128 | |
| 129 def testReturn2(self): | |
| 130 snippet = """ | |
| 131 module testreturn; | |
| 132 function int t() | |
| 133 { | |
| 134 return 2; | |
| 135 } | |
| 136 """ | |
| 137 self.expectOK(snippet) | |
| 138 | |
| 215 | 139 def testExpressions(self): |
| 287 | 140 snippet = """ |
| 284 | 141 module test; |
| 167 | 142 function void t(int a, double b) |
| 143 { | |
| 144 var int a2; | |
| 145 var bool c; | |
| 146 | |
| 147 a2 = b * a; | |
| 148 c = a; | |
| 149 c = b > 1; | |
| 150 } | |
| 287 | 151 """ |
| 152 self.expectErrors(snippet, [8, 9, 10]) | |
| 205 | 153 |
| 230 | 154 def testExpression1(self): |
| 287 | 155 snippet = """ |
| 284 | 156 module testexpr1; |
| 230 | 157 function void t() |
| 158 { | |
| 159 var int a, b, c; | |
| 160 a = 1; | |
| 161 b = a * 2 + a * a; | |
| 162 c = b * a - 3; | |
| 163 } | |
| 287 | 164 """ |
| 165 self.expectOK(snippet) | |
| 230 | 166 |
| 215 | 167 def testEmpty(self): |
| 287 | 168 snippet = """ |
| 169 module A | |
| 170 """ | |
| 171 self.expectErrors(snippet, [3]) | |
| 205 | 172 |
| 215 | 173 def testEmpty2(self): |
| 287 | 174 snippet = "" |
| 175 self.expectErrors(snippet, [1]) | |
| 205 | 176 |
| 215 | 177 def testRedefine(self): |
| 287 | 178 snippet = """ |
| 179 module test; | |
| 180 var int a; | |
| 181 var int b; | |
| 182 var int a; | |
| 183 """ | |
| 184 self.expectErrors(snippet, [5]) | |
| 205 | 185 |
| 215 | 186 def testWhile(self): |
| 287 | 187 snippet = """ |
| 188 module tstwhile; | |
| 189 var int a; | |
| 190 function void t() | |
| 191 { | |
| 306 | 192 var int i; |
| 193 i = 0; | |
|
169
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
194 while (i < 1054) |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
195 { |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
196 i = i + 3; |
| 186 | 197 a = a + i; |
|
169
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
198 } |
| 205 | 199 |
| 200 while(true) | |
| 201 { | |
| 202 } | |
| 203 | |
| 204 while(false) | |
| 205 { | |
| 206 } | |
| 287 | 207 } |
| 208 """ | |
| 209 self.expectOK(snippet) | |
| 215 | 210 |
| 211 def testIf(self): | |
| 212 snippet = """ | |
| 287 | 213 module tstIFF; |
| 214 function void t(int b) | |
| 215 { | |
| 180 | 216 var int a; |
|
169
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
217 a = 2; |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
218 if (a > b) |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
219 { |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
220 if (a > 1337) |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
221 { |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
222 b = 2; |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
223 } |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
224 } |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
225 else |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
226 { |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
227 b = 1; |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
228 } |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
229 |
|
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
168
diff
changeset
|
230 return b; |
| 287 | 231 } |
| 215 | 232 """ |
| 233 self.expectOK(snippet) | |
| 225 | 234 |
| 235 def testTypeDef(self): | |
| 236 snippet = """ | |
| 284 | 237 module testtypedef; |
| 225 | 238 type int my_int; |
| 239 function void t() | |
| 240 { | |
| 241 var my_int a; | |
| 242 var int b; | |
| 243 a = 2; | |
| 244 b = a + 2; | |
| 245 } | |
| 246 """ | |
| 247 self.expectOK(snippet) | |
| 248 | |
| 230 | 249 def testLocalVariable(self): |
| 250 snippet = """ | |
| 284 | 251 module testlocalvar; |
| 230 | 252 function void t() |
| 253 { | |
| 254 var int a, b; | |
| 255 a = 2; | |
| 256 b = a + 2; | |
| 257 } | |
| 258 """ | |
| 268 | 259 self.expectOK(snippet) |
| 230 | 260 |
| 249 | 261 def testUnknownType(self): |
| 284 | 262 snippet = """module testlocalvar; |
| 249 | 263 function void t() |
| 264 { | |
| 265 var int2 a; | |
| 266 } | |
| 267 """ | |
| 268 self.expectErrors(snippet, [4]) | |
| 269 | |
| 230 | 270 def testStruct1(self): |
| 271 snippet = """ | |
| 284 | 272 module teststruct1; |
| 230 | 273 function void t() |
| 274 { | |
| 275 var struct {int x, y;} a; | |
| 276 a.x = 2; | |
| 277 a.y = a.x + 2; | |
| 278 } | |
| 279 """ | |
| 268 | 280 self.expectOK(snippet) |
| 230 | 281 |
| 293 | 282 def testStructCall(self): |
| 283 snippet = """ | |
| 284 module teststruct1; | |
| 285 function void t() | |
| 286 { | |
| 287 var struct {int x, y;} a; | |
| 288 a.b.c(9); | |
| 289 } | |
| 290 """ | |
| 291 self.expectOK(snippet) | |
| 292 | |
| 225 | 293 def testPointerType1(self): |
| 294 snippet = """ | |
| 284 | 295 module testpointer1; |
| 225 | 296 var int* pa; |
| 297 function void t() | |
| 298 { | |
| 299 var int a; | |
| 300 pa = &a; | |
| 301 *pa = 22; | |
| 302 } | |
| 303 """ | |
| 304 self.expectOK(snippet) | |
| 305 | |
| 215 | 306 def testPointerType(self): |
| 213 | 307 snippet = """ |
| 284 | 308 module testpointer; |
| 225 | 309 var int* pa, pb; |
| 213 | 310 function void t(int a, double b) |
| 311 { | |
|
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
312 var int a2; |
|
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
313 a2 = a; // parameters cannot be escaped for now.. |
|
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
272
diff
changeset
|
314 pa = &a2; |
| 225 | 315 pb = pa; |
| 213 | 316 *pa = 22; |
| 317 } | |
| 318 """ | |
| 215 | 319 self.expectOK(snippet) |
| 213 | 320 |
| 221 | 321 def testPointerTypeInCorrect(self): |
| 322 snippet = """ | |
| 284 | 323 module testpointerincorrect; |
| 221 | 324 var int* pa; |
| 325 function void t(int a, double b) | |
| 326 { | |
| 327 pa = 2; // type conflict | |
| 328 pa = &a; | |
| 329 pa = &2; | |
| 330 &a = pa; // No valid lvalue | |
| 331 **pa = 22; // Cannot deref int | |
| 332 } | |
| 333 """ | |
| 228 | 334 self.expectErrors(snippet, [6, 9, 10]) |
| 221 | 335 |
| 230 | 336 def testPointerTypeIr(self): |
| 337 snippet = """ | |
| 284 | 338 module testptr_ir; |
| 230 | 339 function void t() |
| 340 { | |
| 341 var int* a; | |
| 342 a = cast<int*>(40); | |
| 343 *a = 2; | |
| 344 } | |
| 345 """ | |
| 268 | 346 self.expectOK(snippet) |
| 230 | 347 |
| 348 def testPointerTypeIr2(self): | |
| 349 snippet = """ | |
| 284 | 350 module testptr_ir; |
| 230 | 351 type struct {int x,y;}* gpio; |
| 352 function void t() | |
| 353 { | |
| 354 var gpio a; | |
| 355 a = cast<gpio>(40); | |
| 356 a->x = 2; | |
| 357 a->y = a->x - 14; | |
| 358 } | |
| 359 """ | |
| 268 | 360 self.expectOK(snippet) |
| 230 | 361 |
| 362 def testWrongCast(self): | |
| 363 snippet = """ | |
| 284 | 364 module testptr_ir; |
| 230 | 365 type struct {int x,y;}* gpio; |
| 366 function void t() | |
| 367 { | |
| 368 var gpio a; | |
| 369 *cast<gpio>(*a); | |
| 370 } | |
| 371 """ | |
| 372 self.expectErrors(snippet, [7, 7]) | |
| 373 | |
| 215 | 374 def testComplexType(self): |
| 213 | 375 snippet = """ |
| 284 | 376 module testpointer; |
| 213 | 377 type int my_int; |
| 378 | |
| 379 type struct { | |
| 380 int x, y; | |
| 381 } point; | |
| 382 | |
| 383 type struct { | |
| 384 int mem1; | |
| 385 int memb2; | |
| 386 point P1; | |
| 387 } my_struct; | |
| 388 | |
| 389 type my_struct* my_sptr; | |
| 225 | 390 var int* pa; |
| 213 | 391 |
| 227 | 392 function void t(int a, int b, my_sptr x) |
| 213 | 393 { |
| 394 var my_struct *msp; | |
| 395 | |
| 225 | 396 var my_struct u, v; |
| 397 var point *pt; | |
| 398 | |
| 399 pt = &msp->P1; | |
| 213 | 400 msp = x; |
| 225 | 401 *pa = 22 + u.mem1 * v.memb2 - u.P1.x; |
| 213 | 402 x->memb2 = *pa + a * b; |
| 403 | |
| 225 | 404 msp->P1.x = a * x->P1.y; |
| 213 | 405 } |
| 406 """ | |
| 215 | 407 self.expectOK(snippet) |
| 205 | 408 |
| 167 | 409 |
| 410 if __name__ == '__main__': | |
| 243 | 411 unittest.main() |