[Numpy-svn] r5396 - trunk/numpy/core/tests

[numpy-svn at scipy.org]

Author: charris
Date: 2008-07-12 16:46:29 -0500 (Sat, 12 Jul 2008)
New Revision: 5396

Added:
   trunk/numpy/core/tests/test_print.py
Log:
Add basic tests of number str() formatting.


Added: trunk/numpy/core/tests/test_print.py
===================================================================
--- trunk/numpy/core/tests/test_print.py	2008-07-12 21:45:08 UTC (rev 5395)
+++ trunk/numpy/core/tests/test_print.py	2008-07-12 21:46:29 UTC (rev 5396)
@@ -0,0 +1,36 @@
+import numpy as np
+from numpy.testing import *
+
+class TestPrint(TestCase):
+
+    def test_float_types(self) :
+        """ Check formatting.
+
+            This is only for the str function, and only for simple types.
+            The precision of np.float and np.longdouble aren't the same as the
+            python float precision.
+
+        """
+        for t in [np.float, np.double, np.longdouble] :
+            for x in [0, 1,-1, 1e10, 1e20] :
+                assert_equal(str(t(x)), str(float(x)))
+
+    def test_complex_types(self) :
+        """Check formatting.
+
+            This is only for the str function, and only for simple types.
+            The precision of np.float and np.longdouble aren't the same as the
+            python float precision.
+
+        """
+        for t in [np.cfloat, np.cdouble, np.clongdouble] :
+            for x in [0, 1,-1, 1e10, 1e20] :
+                assert_equal(str(t(x)), str(complex(x)))
+                assert_equal(str(t(x*1j)), str(complex(x*1j)))
+                assert_equal(str(t(x + x*1j)), str(complex(x + x*1j)))
+
+
+
+
+if __name__ == "__main__":
+    run_module_suite()