[pypy-svn] r26483 - pypy/dist/pypy/rpython/test

[pedronis at codespeak.net]

Author: pedronis
Date: Fri Apr 28 05:01:04 2006
New Revision: 26483

Modified:
   pypy/dist/pypy/rpython/test/test_rclass.py
Log:
add variation on this test. (remove a left-over print)



Modified: pypy/dist/pypy/rpython/test/test_rclass.py
==============================================================================
--- pypy/dist/pypy/rpython/test/test_rclass.py	(original)
+++ pypy/dist/pypy/rpython/test/test_rclass.py	Fri Apr 28 05:01:04 2006
@@ -58,7 +58,6 @@
         assert res == 12
 
     def test_classattr_both(self):
-        print self.ts
         class A:
             a = 1
         class B(A):
@@ -78,6 +77,28 @@
         res = interpret(dummyfn, [1], type_system=self.ts)
         assert res == 4
 
+    def test_classattr_both2(self):
+        class Base(object):
+            a = 0
+        class A(Base):
+            a = 1
+        class B(Base):
+            a = 2
+        def pick(i):
+            if i == 0:
+                return A
+            else:
+                return B
+            
+        def dummyfn(i):
+            C = pick(i)
+            i = C()
+            return C.a + i.a
+        res = interpret(dummyfn, [0], type_system=self.ts)
+        assert res == 2
+        res = interpret(dummyfn, [1], type_system=self.ts)
+        assert res == 4
+
     def test_classattr_as_defaults(self):
         def dummyfn():
             x = Random()