diff options
Diffstat (limited to 'numpy/core')
| -rw-r--r-- | numpy/core/tests/test_arrayprint.py | 82 | ||||
| -rw-r--r-- | numpy/core/tests/test_regression.py | 2 |
2 files changed, 47 insertions, 37 deletions
diff --git a/numpy/core/tests/test_arrayprint.py b/numpy/core/tests/test_arrayprint.py index 2a3bbcbc3..7ff2efce9 100644 --- a/numpy/core/tests/test_arrayprint.py +++ b/numpy/core/tests/test_arrayprint.py @@ -11,7 +11,7 @@ from numpy.testing import ( class TestArrayRepr(object): def test_nan_inf(self): x = np.array([np.nan, np.inf]) - assert_equal(repr(x), 'array([ nan, inf])') + assert_equal(repr(x), 'array([nan, inf])') def test_subclass(self): class sub(np.ndarray): pass @@ -68,45 +68,45 @@ class TestComplexArray(object): 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]', + '[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]', '[-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]'] + '[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) + assert_equal(res, val) class TestArray2String(object): def test_basic(self): @@ -157,7 +157,7 @@ class TestArray2String(object): dt = np.dtype([('name', np.str_, 16), ('grades', np.float64, (2,))]) x = np.array([('Sarah', (8.0, 7.0)), ('John', (6.0, 7.0))], dtype=dt) assert_equal(np.array2string(x), - "[('Sarah', [ 8., 7.]) ('John', [ 6., 7.])]") + "[('Sarah', [8., 7.]) ('John', [6., 7.])]") # for issue #5692 A = np.zeros(shape=10, dtype=[("A", "M8[s]")]) @@ -180,7 +180,7 @@ class TestArray2String(object): # See #8172 array_scalar = np.array( (1., 2.1234567890123456789, 3.), dtype=('f8,f8,f8')) - assert_equal(np.array2string(array_scalar), "( 1., 2.12345679, 3.)") + assert_equal(np.array2string(array_scalar), "(1., 2.12345679, 3.)") class TestPrintOptions(object): @@ -194,17 +194,17 @@ class TestPrintOptions(object): def test_basic(self): x = np.array([1.5, 0, 1.234567890]) - assert_equal(repr(x), "array([ 1.5 , 0. , 1.23456789])") + assert_equal(repr(x), "array([1.5 , 0. , 1.23456789])") np.set_printoptions(precision=4) - assert_equal(repr(x), "array([ 1.5 , 0. , 1.2346])") + assert_equal(repr(x), "array([1.5 , 0. , 1.2346])") def test_precision_zero(self): np.set_printoptions(precision=0) for values, string in ( - ([0.], " 0."), ([.3], " 0."), ([-.3], "-0."), ([.7], " 1."), - ([1.5], " 2."), ([-1.5], "-2."), ([-15.34], "-15."), - ([100.], " 100."), ([.2, -1, 122.51], " 0., -1., 123."), - ([0], "0"), ([-12], "-12"), ([complex(.3, -.7)], " 0.-1.j")): + ([0.], "0."), ([.3], "0."), ([-.3], "-0."), ([.7], "1."), + ([1.5], "2."), ([-1.5], "-2."), ([-15.34], "-15."), + ([100.], "100."), ([.2, -1, 122.51], " 0., -1., 123."), + ([0], "0"), ([-12], "-12"), ([complex(.3, -.7)], "0.-1.j")): x = np.array(values) assert_equal(repr(x), "array([%s])" % string) @@ -234,7 +234,7 @@ class TestPrintOptions(object): np.set_printoptions(formatter={'float':lambda x: str(x-1)}) assert_equal(repr(x), "array([-1.0, 0.0, 1.0])") np.set_printoptions(formatter={'float_kind':None}) - assert_equal(repr(x), "array([ 0., 1., 2.])") + assert_equal(repr(x), "array([0., 1., 2.])") def test_0d_arrays(self): assert_equal(repr(np.datetime64('2005-02-25')[...]), @@ -248,16 +248,26 @@ class TestPrintOptions(object): x = np.array([1., 2., 3.]) y = np.array([1., 2., -10.]) z = np.array([100., 2., -1.]) + w = np.array([-100., 2., 1.]) assert_equal(repr(x), 'array([1., 2., 3.])') assert_equal(repr(y), 'array([ 1., 2., -10.])') assert_equal(repr(np.array(y[0])), 'array(1.)') assert_equal(repr(np.array(y[-1])), 'array(-10.)') - assert_equal(repr(z), 'array([ 100., 2., -1.])') + assert_equal(repr(z), 'array([100., 2., -1.])') + assert_equal(repr(w), 'array([-100., 2., 1.])') assert_equal(repr(np.array([np.nan, np.inf])), 'array([nan, inf])') assert_equal(repr(np.array([np.nan, -np.inf])), 'array([ nan, -inf])') + x = np.array([np.inf, 100000, 1.1234]) + y = np.array([np.inf, 100000, -1.1234]) + z = np.array([np.inf, 1.1234, -1e120]) + np.set_printoptions(precision=2) + assert_equal(repr(x), 'array([ inf, 1.00e+05, 1.12e+00])') + assert_equal(repr(y), 'array([ inf, 1.00e+05, -1.12e+00])') + assert_equal(repr(z), 'array([ inf, 1.12e+000, -1.00e+120])') + def test_bool_spacing(self): assert_equal(repr(np.array([True, True])), 'array([ True, True], dtype=bool)') diff --git a/numpy/core/tests/test_regression.py b/numpy/core/tests/test_regression.py index 84469d03b..34f9080fb 100644 --- a/numpy/core/tests/test_regression.py +++ b/numpy/core/tests/test_regression.py @@ -858,7 +858,7 @@ class TestRegression(object): def test_sign_bit(self): x = np.array([0, -0.0, 0]) - assert_equal(str(np.abs(x)), '[ 0. 0. 0.]') + assert_equal(str(np.abs(x)), '[0. 0. 0.]') def test_flat_index_byteswap(self): for dt in (np.dtype('<i4'), np.dtype('>i4')): |