|
209
|
1 import unittest
|
|
|
2 import ppci, codegen, ir
|
|
218
|
3 import cortexm3 as arm
|
|
240
|
4 import codegenarm
|
|
|
5 import outstream
|
|
|
6
|
|
|
7 def genTestFunction():
|
|
|
8 m = ir.Module('tst')
|
|
|
9 f = ir.Function('tst')
|
|
|
10 m.addFunction(f)
|
|
|
11 bb = ir.BasicBlock('entry')
|
|
|
12 f.addBasicBlock(bb)
|
|
|
13 return m, bb
|
|
|
14
|
|
209
|
15
|
|
|
16 class testCodeGeneration(unittest.TestCase):
|
|
|
17 def setUp(self):
|
|
218
|
18 self.cg = codegen.CodeGenerator(arm.armtarget)
|
|
209
|
19
|
|
|
20 def testFunction(self):
|
|
240
|
21 m, bb = genTestFunction()
|
|
209
|
22 v = ir.Value('tst')
|
|
|
23 bb.addInstruction(ir.ImmLoad(v, 123))
|
|
|
24 m.check()
|
|
|
25 obj = self.cg.generate(m)
|
|
|
26 self.assertTrue(obj)
|
|
|
27
|
|
240
|
28 class testArmCodeGeneration(unittest.TestCase):
|
|
|
29 def testStack(self):
|
|
|
30 s = outstream.OutputStream()
|
|
|
31 cg = codegenarm.ArmCodeGenerator(s)
|
|
|
32 m, bb = genTestFunction()
|
|
|
33 v = ir.Value('tst')
|
|
|
34 bb.addInstruction(ir.Alloc(v))
|
|
|
35 v2 = ir.Value('tst2')
|
|
|
36 bb.addInstruction(ir.ImmLoad(v2, 22))
|
|
|
37 bb.addInstruction(ir.Store(v, v2))
|
|
|
38 m.check()
|
|
|
39 cg.generate(m)
|
|
|
40 s.dump()
|
|
|
41
|
|
209
|
42 if __name__ == '__main__':
|
|
240
|
43 unittest.main()
|
