1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
| import numpy as np
| from numpy import fix, isneginf, isposinf
| from numpy.testing import assert_, assert_array_equal, assert_equal, assert_raises
|
|
| class TestUfunclike:
|
| def test_isposinf(self):
| a = np.array([np.inf, -np.inf, np.nan, 0.0, 3.0, -3.0])
| out = np.zeros(a.shape, bool)
| tgt = np.array([True, False, False, False, False, False])
|
| res = isposinf(a)
| assert_equal(res, tgt)
| res = isposinf(a, out)
| assert_equal(res, tgt)
| assert_equal(out, tgt)
|
| a = a.astype(np.complex128)
| with assert_raises(TypeError):
| isposinf(a)
|
| def test_isneginf(self):
| a = np.array([np.inf, -np.inf, np.nan, 0.0, 3.0, -3.0])
| out = np.zeros(a.shape, bool)
| tgt = np.array([False, True, False, False, False, False])
|
| res = isneginf(a)
| assert_equal(res, tgt)
| res = isneginf(a, out)
| assert_equal(res, tgt)
| assert_equal(out, tgt)
|
| a = a.astype(np.complex128)
| with assert_raises(TypeError):
| isneginf(a)
|
| def test_fix(self):
| a = np.array([[1.0, 1.1, 1.5, 1.8], [-1.0, -1.1, -1.5, -1.8]])
| out = np.zeros(a.shape, float)
| tgt = np.array([[1., 1., 1., 1.], [-1., -1., -1., -1.]])
|
| res = fix(a)
| assert_equal(res, tgt)
| res = fix(a, out)
| assert_equal(res, tgt)
| assert_equal(out, tgt)
| assert_equal(fix(3.14), 3)
|
| def test_fix_with_subclass(self):
| class MyArray(np.ndarray):
| def __new__(cls, data, metadata=None):
| res = np.array(data, copy=True).view(cls)
| res.metadata = metadata
| return res
|
| def __array_wrap__(self, obj, context=None, return_scalar=False):
| if not isinstance(obj, MyArray):
| obj = obj.view(MyArray)
| if obj.metadata is None:
| obj.metadata = self.metadata
| return obj
|
| def __array_finalize__(self, obj):
| self.metadata = getattr(obj, 'metadata', None)
| return self
|
| a = np.array([1.1, -1.1])
| m = MyArray(a, metadata='foo')
| f = fix(m)
| assert_array_equal(f, np.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 = 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)
|
|
