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

[arigo at codespeak.net]

Author: arigo
Date: Sat May  8 19:28:42 2004
New Revision: 4331

Modified:
   pypy/trunk/src/pypy/translator/test/test_annrpython.py
Log:
More precise test, to ensure that the correct classes get the correct
attributes.


Modified: pypy/trunk/src/pypy/translator/test/test_annrpython.py
==============================================================================
--- pypy/trunk/src/pypy/translator/test/test_annrpython.py	(original)
+++ pypy/trunk/src/pypy/translator/test/test_annrpython.py	Sat May  8 19:28:42 2004
@@ -189,6 +189,15 @@
         self.assertEquals(s.items[1].knowntype, int)
         self.assertEquals(s.items[1].nonneg, True)
 
+    def test_classes_methodcall1(self):
+        a = RPythonAnnotator()
+        a.build_types(snippet.methodcall1, [int])
+        # the user classes should have the following attributes:
+        classes = a.bookkeeper.userclasses
+        self.assertEquals(classes[snippet.F].attrs.keys(), ['m'])
+        self.assertEquals(classes[snippet.G].attrs.keys(), ['m2'])
+        self.assertEquals(classes[snippet.H].attrs,
+                          {'attr': annmodel.immutablevalue(1)})
 
 def g(n):
     return [0,1,2,n]