[pypy-commit] pypy cpyext-injection: improve the test

[fijal]

Author: fijal
Branch: cpyext-injection
Changeset: r87921:7e4d7893385b
Date: 2016-10-24 20:13 +0200
http://bitbucket.org/pypy/pypy/changeset/7e4d7893385b/

Log:	improve the test

diff --git a/pypy/module/cpyext/injection/numpy.py b/pypy/module/cpyext/injection/numpy.py
--- a/pypy/module/cpyext/injection/numpy.py
+++ b/pypy/module/cpyext/injection/numpy.py
@@ -30,10 +30,7 @@
 
 class W_ArrayObject(W_Root):
     def getclass(self, space):
-        import pdb
-        pdb.set_trace()
         return space.fromcache(Original).w_array_type
-        return W_Root.getclass(self, space)
 W_ArrayObject.typedef = TypeDef("ndarray")
 
 class W_Float64Object(W_FloatObject):
diff --git a/pypy/module/cpyext/injection/test/test_numpy.py b/pypy/module/cpyext/injection/test/test_numpy.py
--- a/pypy/module/cpyext/injection/test/test_numpy.py
+++ b/pypy/module/cpyext/injection/test/test_numpy.py
@@ -27,9 +27,12 @@
         array = np.ndarray(100)
         array[10] = 1.0
         assert array[10] == 1.0
+        assert array[10] + array[10] == 2.0
         float64 = np.typeinfo['DOUBLE'][-1]
         assert type(array[10]) is float64
         class X(np.ndarray):
             pass
 
         x = X(1)
+        assert isinstance(x, X)
+        assert isinstance(x, np.ndarray)