FM / tests

/opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests/

/opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests

Name	Size	Mode	Actions
data/	-	0755	rm
__pycache__/	-	0755	rm
test_arraypad.py	54830	0644	edit dl rm
test_arraysetops.py	35912	0644	edit dl rm
test_arrayterator.py	1291	0644	edit dl rm
test_financial_expired.py	247	0644	edit dl rm
test_format.py	41028	0644	edit dl rm
test_function_base.py	157726	0644	edit dl rm
test_histograms.py	32734	0644	edit dl rm
test_index_tricks.py	20256	0644	edit dl rm
test_io.py	107891	0644	edit dl rm
test_loadtxt.py	38560	0644	edit dl rm
test_mixins.py	7030	0644	edit dl rm
test_nanfunctions.py	47609	0644	edit dl rm
test_packbits.py	17546	0644	edit dl rm
test_polynomial.py	11395	0644	edit dl rm
test_recfunctions.py	44001	0644	edit dl rm
test_regression.py	8257	0644	edit dl rm
test_shape_base.py	26817	0644	edit dl rm
test_stride_tricks.py	22849	0644	edit dl rm
test_twodim_base.py	18841	0644	edit dl rm
test_type_check.py	15114	0644	edit dl rm
test_ufunclike.py	2982	0644	edit dl rm
test_utils.py	6218	0644	edit dl rm
test__datasource.py	10571	0644	edit dl rm
test__iotools.py	13743	0644	edit dl rm
test__version.py	1999	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests/test_ufunclike.py (2982B)

import numpy as np
import numpy.core as nx
import numpy.lib.ufunclike as ufl
from numpy.testing import (
    assert_, assert_equal, assert_array_equal, assert_warns, assert_raises
)

class TestUfunclike:

def test_isposinf(self):
        a = nx.array([nx.inf, -nx.inf, nx.nan, 0.0, 3.0, -3.0])
        out = nx.zeros(a.shape, bool)
        tgt = nx.array([True, False, False, False, False, False])

res = ufl.isposinf(a)
        assert_equal(res, tgt)
        res = ufl.isposinf(a, out)
        assert_equal(res, tgt)
        assert_equal(out, tgt)

a = a.astype(np.complex_)
        with assert_raises(TypeError):
            ufl.isposinf(a)

def test_isneginf(self):
        a = nx.array([nx.inf, -nx.inf, nx.nan, 0.0, 3.0, -3.0])
        out = nx.zeros(a.shape, bool)
        tgt = nx.array([False, True, False, False, False, False])

res = ufl.isneginf(a)
        assert_equal(res, tgt)
        res = ufl.isneginf(a, out)
        assert_equal(res, tgt)
        assert_equal(out, tgt)

a = a.astype(np.complex_)
        with assert_raises(TypeError):
            ufl.isneginf(a)

def test_fix(self):
        a = nx.array([[1.0, 1.1, 1.5, 1.8], [-1.0, -1.1, -1.5, -1.8]])
        out = nx.zeros(a.shape, float)
        tgt = nx.array([[1., 1., 1., 1.], [-1., -1., -1., -1.]])

res = ufl.fix(a)
        assert_equal(res, tgt)
        res = ufl.fix(a, out)
        assert_equal(res, tgt)
        assert_equal(out, tgt)
        assert_equal(ufl.fix(3.14), 3)

def test_fix_with_subclass(self):
        class MyArray(nx.ndarray):
            def __new__(cls, data, metadata=None):
                res = nx.array(data, copy=True).view(cls)
                res.metadata = metadata
                return res

def __array_wrap__(self, obj, context=None):
                if isinstance(obj, MyArray):
                    obj.metadata = self.metadata
                return obj

def __array_finalize__(self, obj):
                self.metadata = getattr(obj, 'metadata', None)
                return self

a = nx.array([1.1, -1.1])
        m = MyArray(a, metadata='foo')
        f = ufl.fix(m)
        assert_array_equal(f, nx.array([1, -1]))
        assert_(isinstance(f, MyArray))
        assert_equal(f.metadata, 'foo')

# check 0d arrays don't decay to scalars
        m0d = m[0,...]
        m0d.metadata = 'bar'
        f0d = ufl.fix(m0d)
        assert_(isinstance(f0d, MyArray))
        assert_equal(f0d.metadata, 'bar')

def test_scalar(self):
        x = np.inf
        actual = np.isposinf(x)
        expected = np.True_
        assert_equal(actual, expected)
        assert_equal(type(actual), type(expected))

x = -3.4
        actual = np.fix(x)
        expected = np.float64(-3.0)
        assert_equal(actual, expected)
        assert_equal(type(actual), type(expected))

out = np.array(0.0)
        actual = np.fix(x, out=out)
        assert_(actual is out)

import numpy as np import numpy.core as nx import numpy.lib.ufunclike as ufl from numpy.testing import ( assert_, assert_equal, assert_array_equal, assert_warns, assert_raises ) class TestUfunclike: def test_isposinf(self): a = nx.array([nx.inf, -nx.inf, nx.nan, 0.0, 3.0, -3.0]) out = nx.zeros(a.shape, bool) tgt = nx.array([True, False, False, False, False, False]) res = ufl.isposinf(a) assert_equal(res, tgt) res = ufl.isposinf(a, out) assert_equal(res, tgt) assert_equal(out, tgt) a = a.astype(np.complex_) with assert_raises(TypeError): ufl.isposinf(a) def test_isneginf(self): a = nx.array([nx.inf, -nx.inf, nx.nan, 0.0, 3.0, -3.0]) out = nx.zeros(a.shape, bool) tgt = nx.array([False, True, False, False, False, False]) res = ufl.isneginf(a) assert_equal(res, tgt) res = ufl.isneginf(a, out) assert_equal(res, tgt) assert_equal(out, tgt) a = a.astype(np.complex_) with assert_raises(TypeError): ufl.isneginf(a) def test_fix(self): a = nx.array([[1.0, 1.1, 1.5, 1.8], [-1.0, -1.1, -1.5, -1.8]]) out = nx.zeros(a.shape, float) tgt = nx.array([[1., 1., 1., 1.], [-1., -1., -1., -1.]]) res = ufl.fix(a) assert_equal(res, tgt) res = ufl.fix(a, out) assert_equal(res, tgt) assert_equal(out, tgt) assert_equal(ufl.fix(3.14), 3) def test_fix_with_subclass(self): class MyArray(nx.ndarray): def __new__(cls, data, metadata=None): res = nx.array(data, copy=True).view(cls) res.metadata = metadata return res def __array_wrap__(self, obj, context=None): if isinstance(obj, MyArray): obj.metadata = self.metadata return obj def __array_finalize__(self, obj): self.metadata = getattr(obj, 'metadata', None) return self a = nx.array([1.1, -1.1]) m = MyArray(a, metadata='foo') f = ufl.fix(m) assert_array_equal(f, nx.array([1, -1])) assert_(isinstance(f, MyArray)) assert_equal(f.metadata, 'foo') # check 0d arrays don't decay to scalars m0d = m[0,...] m0d.metadata = 'bar' f0d = ufl.fix(m0d) assert_(isinstance(f0d, MyArray)) assert_equal(f0d.metadata, 'bar') def test_scalar(self): x = np.inf actual = np.isposinf(x) expected = np.True_ assert_equal(actual, expected) assert_equal(type(actual), type(expected)) x = -3.4 actual = np.fix(x) expected = np.float64(-3.0) assert_equal(actual, expected) assert_equal(type(actual), type(expected)) out = np.array(0.0) actual = np.fix(x, out=out) assert_(actual is out)