[pypy-svn] r4918 - pypy/trunk/src/pypy/translator/test

[ale at codespeak.net]

Author: ale
Date: Fri Jun  4 17:16:20 2004
New Revision: 4918

Modified:
   pypy/trunk/src/pypy/translator/test/test_translator.py
Log:
added some tests.

the function call_five doesnt work as inplace operators is not properly handled

Modified: pypy/trunk/src/pypy/translator/test/test_translator.py
==============================================================================
--- pypy/trunk/src/pypy/translator/test/test_translator.py	(original)
+++ pypy/trunk/src/pypy/translator/test/test_translator.py	Fri Jun  4 17:16:20 2004
@@ -9,30 +9,35 @@
 
     def test_set_attr(self):
         t = Translator(snippet.set_attr)
+        t.simplify()
         t.annotate([])
         set_attr = t.compile()
         self.assertEquals(set_attr(), 2)
 
     def test_inheritance2(self):
         t = Translator(snippet.inheritance2)
+        t.simplify()
         t.annotate([])
         inheritance2 = t.compile()
         self.assertEquals(inheritance2(), ((-12, -12), (3, "world")))
 
     def test_factorial2(self):
         t = Translator(snippet.factorial2)
+        t.simplify()
         t.annotate([int])
         factorial2 = t.compile()
         self.assertEquals(factorial2(5), 120)
 
     def test_factorial(self):
         t = Translator(snippet.factorial)
+        t.simplify()
         t.annotate([int])
         factorial = t.compile()
         self.assertEquals(factorial(5), 120)
 
     def test_simple_method(self):
         t = Translator(snippet.simple_method)
+        t.simplify()
         t.annotate([int]).simplify()
         simple_method = t.compile()
         self.assertEquals(simple_method(55), 55)
@@ -45,5 +50,19 @@
         sieve_of_eratosthenes = t.compile()
         self.assertEquals(sieve_of_eratosthenes(), 1028)
 
+    def test_nested_whiles(self):
+        t = Translator(snippet.nested_whiles)
+        t.simplify()
+        t.annotate([int,int])
+        nested_whiles = t.compile()
+        self.assertEquals(nested_whiles(5,3), '!!!!!')
+
+    def test_call_five(self):
+        t = Translator(snippet.call_five)
+        t.simplify()
+        t.annotate([])
+        call_five = t.compile()
+        self.assertEquals(call_five(), [5])
+
 if __name__ == '__main__':
     testit.main()