Author: Taavi Burns <[email protected]>
Branch: 
Changeset: r66134:64bcad26b09c
Date: 2013-08-13 16:43 -0400
http://bitbucket.org/pypy/pypy/changeset/64bcad26b09c/

Log:    Adds a few more tests of types for zeros_like and ones_like.

diff --git a/pypy/module/micronumpy/test/test_ufuncs.py 
b/pypy/module/micronumpy/test/test_ufuncs.py
--- a/pypy/module/micronumpy/test/test_ufuncs.py
+++ b/pypy/module/micronumpy/test/test_ufuncs.py
@@ -956,13 +956,17 @@
         assert logaddexp2(float('inf'), float('inf')) == float('inf')
 
     def test_ones_like(self):
-        from numpypy import array, complex, ones_like
+        from numpypy import array, ones_like
 
+        assert ones_like(False) == array(True)
         assert ones_like(2) == array(1)
+        assert ones_like(2.) == array(1.)
         assert ones_like(complex(2)) == array(complex(1))
 
     def test_zeros_like(self):
-        from numpypy import array, complex, zeros_like
+        from numpypy import array, zeros_like
 
+        assert zeros_like(True) == array(False)
         assert zeros_like(2) == array(0)
+        assert zeros_like(2.) == array(0.)
         assert zeros_like(complex(2)) == array(complex(0))
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit

Reply via email to