FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
python/test/numpy/ndarray.py at python313 · boostorg/python · GitHub
Uh oh!
There was an error while loading.
Please reload this page
.
boostorg
/
python
Public
Notifications
You must be signed in to change notification settings
Fork
223
Star
537
Code
Issues
173
Pull requests
34
Actions
Projects
Wiki
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Wiki
Security and quality
Insights
Expand file tree
Breadcrumbs
python
/
test
/
numpy
/
ndarray.py
Copy path
More file actions
More file actions
Latest commit
History
History
History
112 lines (98 loc) · 3.88 KB
Breadcrumbs
python
/
test
/
numpy
/
ndarray.py
Copy path
File metadata and controls
112 lines (98 loc) · 3.88 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#!/usr/bin/env python3
# Copyright Jim Bosch & Ankit Daftery 2010-2012.
# Distributed under the Boost Software License, Version 1.0.
# (See accompanying file LICENSE_1_0.txt or copy at
# http://www.boost.org/LICENSE_1_0.txt)
import
ndarray_ext
import
unittest
import
numpy
class
TestNdarray
(
unittest
.
TestCase
):
def
testNdzeros
(
self
):
for
dtp
in
(
numpy
.
int16
,
numpy
.
int32
,
numpy
.
float32
,
numpy
.
complex128
):
v
=
numpy
.
zeros
(
60
,
dtype
=
dtp
)
dt
=
numpy
.
dtype
(
dtp
)
for
shape
in
((
60
,),(
6
,
10
),(
4
,
3
,
5
),(
2
,
2
,
3
,
5
)):
a1
=
ndarray_ext
.
zeros
(
shape
,
dt
)
a2
=
v
.
reshape
(
a1
.
shape
)
self
.
assertEqual
(
shape
,
a1
.
shape
)
self
.
assertTrue
((
a1
==
a2
).
all
())
def
testNdzeros_matrix
(
self
):
for
dtp
in
(
numpy
.
int16
,
numpy
.
int32
,
numpy
.
float32
,
numpy
.
complex128
):
dt
=
numpy
.
dtype
(
dtp
)
shape
=
(
6
,
10
)
a1
=
ndarray_ext
.
zeros_matrix
(
shape
,
dt
)
a2
=
numpy
.
matrix
(
numpy
.
zeros
(
shape
,
dtype
=
dtp
))
self
.
assertEqual
(
shape
,
a1
.
shape
)
self
.
assertTrue
((
a1
==
a2
).
all
())
self
.
assertEqual
(
type
(
a1
),
type
(
a2
))
def
testNdarray
(
self
):
a
=
range
(
0
,
60
)
for
dtp
in
(
numpy
.
int16
,
numpy
.
int32
,
numpy
.
float32
,
numpy
.
complex128
):
v
=
numpy
.
array
(
a
,
dtype
=
dtp
)
dt
=
numpy
.
dtype
(
dtp
)
a1
=
ndarray_ext
.
array
(
a
)
a2
=
ndarray_ext
.
array
(
a
,
dt
)
self
.
assertTrue
((
a1
==
v
).
all
())
self
.
assertTrue
((
a2
==
v
).
all
())
for
shape
in
((
60
,),(
6
,
10
),(
4
,
3
,
5
),(
2
,
2
,
3
,
5
)):
a1
=
a1
.
reshape
(
shape
)
self
.
assertEqual
(
shape
,
a1
.
shape
)
a2
=
a2
.
reshape
(
shape
)
self
.
assertEqual
(
shape
,
a2
.
shape
)
def
testNdempty
(
self
):
for
dtp
in
(
numpy
.
int16
,
numpy
.
int32
,
numpy
.
float32
,
numpy
.
complex128
):
dt
=
numpy
.
dtype
(
dtp
)
for
shape
in
((
60
,),(
6
,
10
),(
4
,
3
,
5
),(
2
,
2
,
3
,
5
)):
a1
=
ndarray_ext
.
empty
(
shape
,
dt
)
a2
=
ndarray_ext
.
c_empty
(
shape
,
dt
)
self
.
assertEqual
(
shape
,
a1
.
shape
)
self
.
assertEqual
(
shape
,
a2
.
shape
)
def
testTranspose
(
self
):
for
dtp
in
(
numpy
.
int16
,
numpy
.
int32
,
numpy
.
float32
,
numpy
.
complex128
):
dt
=
numpy
.
dtype
(
dtp
)
for
shape
in
((
6
,
10
),(
4
,
3
,
5
),(
2
,
2
,
3
,
5
)):
a1
=
numpy
.
empty
(
shape
,
dt
)
a2
=
a1
.
transpose
()
a1
=
ndarray_ext
.
transpose
(
a1
)
self
.
assertEqual
(
a1
.
shape
,
a2
.
shape
)
def
testSqueeze
(
self
):
a1
=
numpy
.
array
([[[
3
,
4
,
5
]]])
a2
=
a1
.
squeeze
()
a1
=
ndarray_ext
.
squeeze
(
a1
)
self
.
assertEqual
(
a1
.
shape
,
a2
.
shape
)
def
testReshape
(
self
):
a1
=
numpy
.
empty
((
2
,
2
))
a2
=
ndarray_ext
.
reshape
(
a1
,(
1
,
4
))
self
.
assertEqual
(
a2
.
shape
,(
1
,
4
))
def
testShapeIndex
(
self
):
a
=
numpy
.
arange
(
24
)
a
.
shape
=
(
1
,
2
,
3
,
4
)
def
shape_check
(
i
):
print
(
i
)
self
.
assertEqual
(
ndarray_ext
.
shape_index
(
a
,
i
) ,
a
.
shape
[
i
] )
for
i
in
range
(
4
):
shape_check
(
i
)
for
i
in
range
(
-
1
,
-
5
,
-
1
):
shape_check
(
i
)
try
:
ndarray_ext
.
shape_index
(
a
,
4
)
# out of bounds -- should raise IndexError
self
.
assertTrue
(
False
)
except
IndexError
:
pass
def
testStridesIndex
(
self
):
a
=
numpy
.
arange
(
24
)
a
.
shape
=
(
1
,
2
,
3
,
4
)
def
strides_check
(
i
):
print
(
i
)
self
.
assertEqual
(
ndarray_ext
.
strides_index
(
a
,
i
) ,
a
.
strides
[
i
] )
for
i
in
range
(
4
):
strides_check
(
i
)
for
i
in
range
(
-
1
,
-
5
,
-
1
):
strides_check
(
i
)
try
:
ndarray_ext
.
strides_index
(
a
,
4
)
# out of bounds -- should raise IndexError
self
.
assertTrue
(
False
)
except
IndexError
:
pass
if
__name__
==
"__main__"
:
unittest
.
main
()
Back
|
FazBrowse Home
|
New Git URL