Gecko [ fernandoquevedodop.com ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
data	[ DIR ]	drwxr-xr-x
__init__.py	0 B	-rw-r--r--
test__datasource.py	10.24 KB	-rw-r--r--
test__iotools.py	13.42 KB	-rw-r--r--
test__version.py	1.95 KB	-rw-r--r--
test_arraypad.py	53.01 KB	-rw-r--r--
test_arraysetops.py	27.76 KB	-rw-r--r--
test_arrayterator.py	1.26 KB	-rw-r--r--
test_financial_expired.py	358 B	-rw-r--r--
test_format.py	37.34 KB	-rw-r--r--
test_function_base.py	132.69 KB	-rw-r--r--
test_histograms.py	32.88 KB	-rw-r--r--
test_index_tricks.py	18.53 KB	-rw-r--r--
test_io.py	100.53 KB	-rw-r--r--
test_mixins.py	6.87 KB	-rw-r--r--
test_nanfunctions.py	37.68 KB	-rw-r--r--
test_packbits.py	17.13 KB	-rw-r--r--
test_polynomial.py	10.46 KB	-rw-r--r--
test_recfunctions.py	40.19 KB	-rw-r--r--
test_regression.py	8.08 KB	-rw-r--r--
test_shape_base.py	23.73 KB	-rw-r--r--
test_stride_tricks.py	22.31 KB	-rw-r--r--
test_twodim_base.py	17.93 KB	-rw-r--r--
test_type_check.py	14.76 KB	-rw-r--r--
test_ufunclike.py	3.2 KB	-rw-r--r--
test_utils.py	4.45 KB	-rw-r--r--

Code Editor : test_ufunclike.py

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_deprecated(self):
        # NumPy 1.13.0, 2017-04-26
        assert_warns(DeprecationWarning, ufl.fix, [1, 2], y=nx.empty(2))
        assert_warns(DeprecationWarning, ufl.isposinf, [1, 2], y=nx.empty(2))
        assert_warns(DeprecationWarning, ufl.isneginf, [1, 2], y=nx.empty(2))

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)

${this.title}

Code Editor : test_ufunclike.py