Chris@87: from __future__ import division, absolute_import, print_function Chris@87: Chris@87: import os Chris@87: import sys Chris@87: Chris@87: import numpy as np Chris@87: from numpy.testing import ( Chris@87: run_module_suite, TestCase, assert_, assert_equal, assert_array_equal, Chris@87: assert_array_almost_equal, assert_raises Chris@87: ) Chris@87: from numpy.testing.utils import _assert_valid_refcount Chris@87: from numpy.compat import unicode Chris@87: Chris@87: rlevel = 1 Chris@87: Chris@87: Chris@87: class TestRegression(TestCase): Chris@87: def test_poly1d(self, level=rlevel): Chris@87: # Ticket #28 Chris@87: assert_equal(np.poly1d([1]) - np.poly1d([1, 0]), Chris@87: np.poly1d([-1, 1])) Chris@87: Chris@87: def test_cov_parameters(self, level=rlevel): Chris@87: # Ticket #91 Chris@87: x = np.random.random((3, 3)) Chris@87: y = x.copy() Chris@87: np.cov(x, rowvar=1) Chris@87: np.cov(y, rowvar=0) Chris@87: assert_array_equal(x, y) Chris@87: Chris@87: def test_mem_digitize(self, level=rlevel): Chris@87: # Ticket #95 Chris@87: for i in range(100): Chris@87: np.digitize([1, 2, 3, 4], [1, 3]) Chris@87: np.digitize([0, 1, 2, 3, 4], [1, 3]) Chris@87: Chris@87: def test_unique_zero_sized(self, level=rlevel): Chris@87: # Ticket #205 Chris@87: assert_array_equal([], np.unique(np.array([]))) Chris@87: Chris@87: def test_mem_vectorise(self, level=rlevel): Chris@87: # Ticket #325 Chris@87: vt = np.vectorize(lambda *args: args) Chris@87: vt(np.zeros((1, 2, 1)), np.zeros((2, 1, 1)), np.zeros((1, 1, 2))) Chris@87: vt(np.zeros((1, 2, 1)), np.zeros((2, 1, 1)), np.zeros((1, Chris@87: 1, 2)), np.zeros((2, 2))) Chris@87: Chris@87: def test_mgrid_single_element(self, level=rlevel): Chris@87: # Ticket #339 Chris@87: assert_array_equal(np.mgrid[0:0:1j], [0]) Chris@87: assert_array_equal(np.mgrid[0:0], []) Chris@87: Chris@87: def test_refcount_vectorize(self, level=rlevel): Chris@87: # Ticket #378 Chris@87: def p(x, y): Chris@87: return 123 Chris@87: v = np.vectorize(p) Chris@87: _assert_valid_refcount(v) Chris@87: Chris@87: def test_poly1d_nan_roots(self, level=rlevel): Chris@87: # Ticket #396 Chris@87: p = np.poly1d([np.nan, np.nan, 1], r=0) Chris@87: self.assertRaises(np.linalg.LinAlgError, getattr, p, "r") Chris@87: Chris@87: def test_mem_polymul(self, level=rlevel): Chris@87: # Ticket #448 Chris@87: np.polymul([], [1.]) Chris@87: Chris@87: def test_mem_string_concat(self, level=rlevel): Chris@87: # Ticket #469 Chris@87: x = np.array([]) Chris@87: np.append(x, 'asdasd\tasdasd') Chris@87: Chris@87: def test_poly_div(self, level=rlevel): Chris@87: # Ticket #553 Chris@87: u = np.poly1d([1, 2, 3]) Chris@87: v = np.poly1d([1, 2, 3, 4, 5]) Chris@87: q, r = np.polydiv(u, v) Chris@87: assert_equal(q*v + r, u) Chris@87: Chris@87: def test_poly_eq(self, level=rlevel): Chris@87: # Ticket #554 Chris@87: x = np.poly1d([1, 2, 3]) Chris@87: y = np.poly1d([3, 4]) Chris@87: assert_(x != y) Chris@87: assert_(x == x) Chris@87: Chris@87: def test_mem_insert(self, level=rlevel): Chris@87: # Ticket #572 Chris@87: np.lib.place(1, 1, 1) Chris@87: Chris@87: def test_polyfit_build(self): Chris@87: # Ticket #628 Chris@87: ref = [-1.06123820e-06, 5.70886914e-04, -1.13822012e-01, Chris@87: 9.95368241e+00, -3.14526520e+02] Chris@87: x = [90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, Chris@87: 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, Chris@87: 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 129, Chris@87: 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, Chris@87: 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, Chris@87: 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, Chris@87: 170, 171, 172, 173, 174, 175, 176] Chris@87: y = [9.0, 3.0, 7.0, 4.0, 4.0, 8.0, 6.0, 11.0, 9.0, 8.0, 11.0, 5.0, Chris@87: 6.0, 5.0, 9.0, 8.0, 6.0, 10.0, 6.0, 10.0, 7.0, 6.0, 6.0, 6.0, Chris@87: 13.0, 4.0, 9.0, 11.0, 4.0, 5.0, 8.0, 5.0, 7.0, 7.0, 6.0, 12.0, Chris@87: 7.0, 7.0, 9.0, 4.0, 12.0, 6.0, 6.0, 4.0, 3.0, 9.0, 8.0, 8.0, Chris@87: 6.0, 7.0, 9.0, 10.0, 6.0, 8.0, 4.0, 7.0, 7.0, 10.0, 8.0, 8.0, Chris@87: 6.0, 3.0, 8.0, 4.0, 5.0, 7.0, 8.0, 6.0, 6.0, 4.0, 12.0, 9.0, Chris@87: 8.0, 8.0, 8.0, 6.0, 7.0, 4.0, 4.0, 5.0, 7.0] Chris@87: tested = np.polyfit(x, y, 4) Chris@87: assert_array_almost_equal(ref, tested) Chris@87: Chris@87: def test_polydiv_type(self): Chris@87: # Make polydiv work for complex types Chris@87: msg = "Wrong type, should be complex" Chris@87: x = np.ones(3, dtype=np.complex) Chris@87: q, r = np.polydiv(x, x) Chris@87: assert_(q.dtype == np.complex, msg) Chris@87: msg = "Wrong type, should be float" Chris@87: x = np.ones(3, dtype=np.int) Chris@87: q, r = np.polydiv(x, x) Chris@87: assert_(q.dtype == np.float, msg) Chris@87: Chris@87: def test_histogramdd_too_many_bins(self): Chris@87: # Ticket 928. Chris@87: assert_raises(ValueError, np.histogramdd, np.ones((1, 10)), bins=2**10) Chris@87: Chris@87: def test_polyint_type(self): Chris@87: # Ticket #944 Chris@87: msg = "Wrong type, should be complex" Chris@87: x = np.ones(3, dtype=np.complex) Chris@87: assert_(np.polyint(x).dtype == np.complex, msg) Chris@87: msg = "Wrong type, should be float" Chris@87: x = np.ones(3, dtype=np.int) Chris@87: assert_(np.polyint(x).dtype == np.float, msg) Chris@87: Chris@87: def test_ndenumerate_crash(self): Chris@87: # Ticket 1140 Chris@87: # Shouldn't crash: Chris@87: list(np.ndenumerate(np.array([[]]))) Chris@87: Chris@87: def test_asfarray_none(self, level=rlevel): Chris@87: # Test for changeset r5065 Chris@87: assert_array_equal(np.array([np.nan]), np.asfarray([None])) Chris@87: Chris@87: def test_large_fancy_indexing(self, level=rlevel): Chris@87: # Large enough to fail on 64-bit. Chris@87: nbits = np.dtype(np.intp).itemsize * 8 Chris@87: thesize = int((2**nbits)**(1.0/5.0)+1) Chris@87: Chris@87: def dp(): Chris@87: n = 3 Chris@87: a = np.ones((n,)*5) Chris@87: i = np.random.randint(0, n, size=thesize) Chris@87: a[np.ix_(i, i, i, i, i)] = 0 Chris@87: Chris@87: def dp2(): Chris@87: n = 3 Chris@87: a = np.ones((n,)*5) Chris@87: i = np.random.randint(0, n, size=thesize) Chris@87: a[np.ix_(i, i, i, i, i)] Chris@87: Chris@87: self.assertRaises(ValueError, dp) Chris@87: self.assertRaises(ValueError, dp2) Chris@87: Chris@87: def test_void_coercion(self, level=rlevel): Chris@87: dt = np.dtype([('a', 'f4'), ('b', 'i4')]) Chris@87: x = np.zeros((1,), dt) Chris@87: assert_(np.r_[x, x].dtype == dt) Chris@87: Chris@87: def test_who_with_0dim_array(self, level=rlevel): Chris@87: # ticket #1243 Chris@87: import os Chris@87: import sys Chris@87: Chris@87: oldstdout = sys.stdout Chris@87: sys.stdout = open(os.devnull, 'w') Chris@87: try: Chris@87: try: Chris@87: np.who({'foo': np.array(1)}) Chris@87: except: Chris@87: raise AssertionError("ticket #1243") Chris@87: finally: Chris@87: sys.stdout.close() Chris@87: sys.stdout = oldstdout Chris@87: Chris@87: def test_include_dirs(self): Chris@87: # As a sanity check, just test that get_include Chris@87: # includes something reasonable. Somewhat Chris@87: # related to ticket #1405. Chris@87: include_dirs = [np.get_include()] Chris@87: for path in include_dirs: Chris@87: assert_(isinstance(path, (str, unicode))) Chris@87: assert_(path != '') Chris@87: Chris@87: def test_polyder_return_type(self): Chris@87: # Ticket #1249 Chris@87: assert_(isinstance(np.polyder(np.poly1d([1]), 0), np.poly1d)) Chris@87: assert_(isinstance(np.polyder([1], 0), np.ndarray)) Chris@87: assert_(isinstance(np.polyder(np.poly1d([1]), 1), np.poly1d)) Chris@87: assert_(isinstance(np.polyder([1], 1), np.ndarray)) Chris@87: Chris@87: def test_append_fields_dtype_list(self): Chris@87: # Ticket #1676 Chris@87: from numpy.lib.recfunctions import append_fields Chris@87: Chris@87: base = np.array([1, 2, 3], dtype=np.int32) Chris@87: names = ['a', 'b', 'c'] Chris@87: data = np.eye(3).astype(np.int32) Chris@87: dlist = [np.float64, np.int32, np.int32] Chris@87: try: Chris@87: append_fields(base, names, data, dlist) Chris@87: except: Chris@87: raise AssertionError() Chris@87: Chris@87: def test_loadtxt_fields_subarrays(self): Chris@87: # For ticket #1936 Chris@87: if sys.version_info[0] >= 3: Chris@87: from io import StringIO Chris@87: else: Chris@87: from StringIO import StringIO Chris@87: Chris@87: dt = [("a", 'u1', 2), ("b", 'u1', 2)] Chris@87: x = np.loadtxt(StringIO("0 1 2 3"), dtype=dt) Chris@87: assert_equal(x, np.array([((0, 1), (2, 3))], dtype=dt)) Chris@87: Chris@87: dt = [("a", [("a", 'u1', (1, 3)), ("b", 'u1')])] Chris@87: x = np.loadtxt(StringIO("0 1 2 3"), dtype=dt) Chris@87: assert_equal(x, np.array([(((0, 1, 2), 3),)], dtype=dt)) Chris@87: Chris@87: dt = [("a", 'u1', (2, 2))] Chris@87: x = np.loadtxt(StringIO("0 1 2 3"), dtype=dt) Chris@87: assert_equal(x, np.array([(((0, 1), (2, 3)),)], dtype=dt)) Chris@87: Chris@87: dt = [("a", 'u1', (2, 3, 2))] Chris@87: x = np.loadtxt(StringIO("0 1 2 3 4 5 6 7 8 9 10 11"), dtype=dt) Chris@87: data = [((((0, 1), (2, 3), (4, 5)), ((6, 7), (8, 9), (10, 11))),)] Chris@87: assert_equal(x, np.array(data, dtype=dt)) Chris@87: Chris@87: def test_nansum_with_boolean(self): Chris@87: # gh-2978 Chris@87: a = np.zeros(2, dtype=np.bool) Chris@87: try: Chris@87: np.nansum(a) Chris@87: except: Chris@87: raise AssertionError() Chris@87: Chris@87: def test_py3_compat(self): Chris@87: # gh-2561 Chris@87: # Test if the oldstyle class test is bypassed in python3 Chris@87: class C(): Chris@87: """Old-style class in python2, normal class in python3""" Chris@87: pass Chris@87: Chris@87: out = open(os.devnull, 'w') Chris@87: try: Chris@87: np.info(C(), output=out) Chris@87: except AttributeError: Chris@87: raise AssertionError() Chris@87: finally: Chris@87: out.close() Chris@87: Chris@87: Chris@87: if __name__ == "__main__": Chris@87: run_module_suite()