Add basic tests of number str() formatting.

1 files changed, 36 insertions, 0 deletions

diff --git a/numpy/core/tests/test_print.py b/numpy/core/tests/test_print.py
new file mode 100644
index 000000000..b4b12048a
--- /dev/null
+++ b/numpy/core/tests/test_print.py
@@ -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()