diff options
| -rw-r--r-- | numpy/core/tests/test_arrayprint.py | 82 |
1 files changed, 46 insertions, 36 deletions
diff --git a/numpy/core/tests/test_arrayprint.py b/numpy/core/tests/test_arrayprint.py index c7b69a09f..d6e2da1ad 100644 --- a/numpy/core/tests/test_arrayprint.py +++ b/numpy/core/tests/test_arrayprint.py @@ -13,45 +13,55 @@ class TestComplexArray(TestCase): dtypes = [np.complex64, np.cdouble, np.clongdouble] actual = [str(np.array([c], dt)) for c in cvals for dt in dtypes] wanted = [ - '[ 0.+0.j]', '[ 0.+0.j]', '[ 0.0+0.0j]', - '[ 0.+1.j]', '[ 0.+1.j]', '[ 0.0+1.0j]', - '[ 0.-1.j]', '[ 0.-1.j]', '[ 0.0-1.0j]', - '[ 0.+infj]', '[ 0.+infj]', '[ 0.0+infj]', - '[ 0.-infj]', '[ 0.-infj]', '[ 0.0-infj]', - '[ 0.+nanj]', '[ 0.+nanj]', '[ 0.0+nanj]', - '[ 1.+0.j]', '[ 1.+0.j]', '[ 1.0+0.0j]', - '[ 1.+1.j]', '[ 1.+1.j]', '[ 1.0+1.0j]', - '[ 1.-1.j]', '[ 1.-1.j]', '[ 1.0-1.0j]', - '[ 1.+infj]', '[ 1.+infj]', '[ 1.0+infj]', - '[ 1.-infj]', '[ 1.-infj]', '[ 1.0-infj]', - '[ 1.+nanj]', '[ 1.+nanj]', '[ 1.0+nanj]', - '[-1.+0.j]', '[-1.+0.j]', '[-1.0+0.0j]', - '[-1.+1.j]', '[-1.+1.j]', '[-1.0+1.0j]', - '[-1.-1.j]', '[-1.-1.j]', '[-1.0-1.0j]', - '[-1.+infj]', '[-1.+infj]', '[-1.0+infj]', - '[-1.-infj]', '[-1.-infj]', '[-1.0-infj]', - '[-1.+nanj]', '[-1.+nanj]', '[-1.0+nanj]', - '[ inf+0.j]', '[ inf+0.j]', '[ inf+0.0j]', - '[ inf+1.j]', '[ inf+1.j]', '[ inf+1.0j]', - '[ inf-1.j]', '[ inf-1.j]', '[ inf-1.0j]', - '[ inf+infj]', '[ inf+infj]', '[ inf+infj]', - '[ inf-infj]', '[ inf-infj]', '[ inf-infj]', - '[ inf+nanj]', '[ inf+nanj]', '[ inf+nanj]', - '[-inf+0.j]', '[-inf+0.j]', '[-inf+0.0j]', - '[-inf+1.j]', '[-inf+1.j]', '[-inf+1.0j]', - '[-inf-1.j]', '[-inf-1.j]', '[-inf-1.0j]', - '[-inf+infj]', '[-inf+infj]', '[-inf+infj]', - '[-inf-infj]', '[-inf-infj]', '[-inf-infj]', - '[-inf+nanj]', '[-inf+nanj]', '[-inf+nanj]', - '[ nan+0.j]', '[ nan+0.j]', '[ nan+0.0j]', - '[ nan+1.j]', '[ nan+1.j]', '[ nan+1.0j]', - '[ nan-1.j]', '[ nan-1.j]', '[ nan-1.0j]', - '[ nan+infj]', '[ nan+infj]', '[ nan+infj]', - '[ nan-infj]', '[ nan-infj]', '[ nan-infj]', + '[ 0.+0.j]', '[ 0.+0.j]', '[ 0.0+0.0j]', + '[ 0.+1.j]', '[ 0.+1.j]', '[ 0.0+1.0j]', + '[ 0.-1.j]', '[ 0.-1.j]', '[ 0.0-1.0j]', + '[ 0.+infj]', '[ 0.+infj]', '[ 0.0+infj]', + '[ 0.-infj]', '[ 0.-infj]', '[ 0.0-infj]', + '[ 0.+nanj]', '[ 0.+nanj]', '[ 0.0+nanj]', + '[ 1.+0.j]', '[ 1.+0.j]', '[ 1.0+0.0j]', + '[ 1.+1.j]', '[ 1.+1.j]', '[ 1.0+1.0j]', + '[ 1.-1.j]', '[ 1.-1.j]', '[ 1.0-1.0j]', + '[ 1.+infj]', '[ 1.+infj]', '[ 1.0+infj]', + '[ 1.-infj]', '[ 1.-infj]', '[ 1.0-infj]', + '[ 1.+nanj]', '[ 1.+nanj]', '[ 1.0+nanj]', + '[-1.+0.j]', '[-1.+0.j]', '[-1.0+0.0j]', + '[-1.+1.j]', '[-1.+1.j]', '[-1.0+1.0j]', + '[-1.-1.j]', '[-1.-1.j]', '[-1.0-1.0j]', + '[-1.+infj]', '[-1.+infj]', '[-1.0+infj]', + '[-1.-infj]', '[-1.-infj]', '[-1.0-infj]', + '[-1.+nanj]', '[-1.+nanj]', '[-1.0+nanj]', + '[ inf+0.j]', '[ inf+0.j]', '[ inf+0.0j]', + '[ inf+1.j]', '[ inf+1.j]', '[ inf+1.0j]', + '[ inf-1.j]', '[ inf-1.j]', '[ inf-1.0j]', + '[ inf+infj]', '[ inf+infj]', '[ inf+infj]', + '[ inf-infj]', '[ inf-infj]', '[ inf-infj]', + '[ inf+nanj]', '[ inf+nanj]', '[ inf+nanj]', + '[-inf+0.j]', '[-inf+0.j]', '[-inf+0.0j]', + '[-inf+1.j]', '[-inf+1.j]', '[-inf+1.0j]', + '[-inf-1.j]', '[-inf-1.j]', '[-inf-1.0j]', + '[-inf+infj]', '[-inf+infj]', '[-inf+infj]', + '[-inf-infj]', '[-inf-infj]', '[-inf-infj]', + '[-inf+nanj]', '[-inf+nanj]', '[-inf+nanj]', + '[ nan+0.j]', '[ nan+0.j]', '[ nan+0.0j]', + '[ nan+1.j]', '[ nan+1.j]', '[ nan+1.0j]', + '[ nan-1.j]', '[ nan-1.j]', '[ nan-1.0j]', + '[ nan+infj]', '[ nan+infj]', '[ nan+infj]', + '[ nan-infj]', '[ nan-infj]', '[ nan-infj]', '[ nan+nanj]', '[ nan+nanj]', '[ nan+nanj]'] for res, val in zip(actual, wanted): assert_(res == val) - + +def test_array2string(): + """Basic test of array2string.""" + a = np.arange(3) + assert_(np.array2string(a) == '[0 1 2]') + assert_(np.array2string(a, max_line_width=4) == '[0 1\n 2]') + stylestr = np.array2string(np.array(1.5), + style=lambda x: "Value in 0-D array: " + str(x)) + assert_(stylestr == 'Value in 0-D array: 1.5') + + if __name__ == "__main__": run_module_suite() |