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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
|
! { dg-do run }
! Transformational functions for zero-sized array and array sections
! Contributed by Francois-Xavier Coudert <coudert@clipper.ens.fr>
subroutine test_cshift
real :: tempn(1), tempm(1,2)
real,allocatable :: foo(:),bar(:,:),gee(:,:)
tempn = 2.0
tempm = 1.0
allocate(foo(0),bar(2,0),gee(0,7))
if (any(cshift(foo,dim=1,shift=1)/= 0)) STOP 1
if (any(cshift(tempn(2:1),dim=1,shift=1)/= 0)) STOP 2
if (any(cshift(bar,shift=(/1,-1/),dim=1)/= 0)) STOP 3
if (any(cshift(bar,shift=(/1,-1/),dim=2)/= 0)) STOP 4
if (any(cshift(gee,shift=(/1,-1/),dim=1)/= 0)) STOP 5
if (any(cshift(gee,shift=(/1,-1/),dim=2)/= 0)) STOP 6
if (any(cshift(tempm(5:4,:),shift=(/1,-1/),dim=1)/= 0)) STOP 7
deallocate(foo,bar,gee)
end
subroutine test_eoshift
real :: tempn(1), tempm(1,2)
real,allocatable :: foo(:),bar(:,:),gee(:,:)
tempn = 2.0
tempm = 1.0
allocate(foo(0),bar(2,0),gee(0,7))
if (any(eoshift(foo,dim=1,shift=1)/= 0)) STOP 8
if (any(eoshift(tempn(2:1),dim=1,shift=1)/= 0)) STOP 9
if (any(eoshift(bar,shift=(/1,-1/),dim=1)/= 0)) STOP 10
if (any(eoshift(bar,shift=(/1,-1/),dim=2)/= 0)) STOP 11
if (any(eoshift(gee,shift=(/1,-1/),dim=1)/= 0)) STOP 12
if (any(eoshift(gee,shift=(/1,-1/),dim=2)/= 0)) STOP 13
if (any(eoshift(tempm(5:4,:),shift=(/1,-1/),dim=1)/= 0)) STOP 14
if (any(eoshift(foo,dim=1,shift=1,boundary=42.0)/= 0)) STOP 15
if (any(eoshift(tempn(2:1),dim=1,shift=1,boundary=42.0)/= 0)) STOP 16
if (any(eoshift(bar,shift=(/1,-1/),dim=1,boundary=42.0)/= 0)) STOP 17
if (any(eoshift(bar,shift=(/1,-1/),dim=2,boundary=42.0)/= 0)) STOP 18
if (any(eoshift(gee,shift=(/1,-1/),dim=1,boundary=42.0)/= 0)) STOP 19
if (any(eoshift(gee,shift=(/1,-1/),dim=2,boundary=42.0)/= 0)) STOP 20
if (any(eoshift(tempm(5:4,:),shift=(/1,-1/),dim=1,boundary=42.0)/= 0)) STOP 21
if (any(eoshift(foo,dim=1,shift=1,boundary=42.0)/= 0)) STOP 22
if (any(eoshift(tempn(2:1),dim=1,shift=1,boundary=-7.0)/= 0)) STOP 23
if (any(eoshift(bar,shift=(/1,-1/),dim=1,boundary=(/42.0,-7.0/))/= 0)) STOP 24
if (any(eoshift(bar,shift=(/1,-1/),dim=2,boundary=(/42.0,-7.0/))/= 0)) STOP 25
if (any(eoshift(gee,shift=(/1,-1/),dim=1,boundary=(/42.0,-7.0/))/= 0)) STOP 26
if (any(eoshift(gee,shift=(/1,-1/),dim=2,boundary=(/42.0,-7.0/))/= 0)) STOP 27
if (any(eoshift(tempm(5:4,:),shift=(/1,-1/),dim=1,boundary=(/42.0,-7.0/))/= 0)) STOP 28
deallocate(foo,bar,gee)
end
subroutine test_transpose
character(len=1) :: tempn(1,2)
character(len=1),allocatable :: foo(:,:), bar(:,:)
integer :: tempm(1,2)
integer,allocatable :: x(:,:), y(:,:)
tempn = 'a'
allocate(foo(3,0),bar(-2:-4,7:9))
tempm = -42
allocate(x(3,0),y(-2:-4,7:9))
if (any(transpose(tempn(-7:-8,:)) /= 'b')) STOP 29
if (any(transpose(tempn(:,9:8)) /= 'b')) STOP 30
if (any(transpose(foo) /= 'b')) STOP 31
if (any(transpose(bar) /= 'b')) STOP 32
if (any(transpose(tempm(-7:-8,:)) /= 0)) STOP 33
if (any(transpose(tempm(:,9:8)) /= 0)) STOP 34
if (any(transpose(x) /= 0)) STOP 35
if (any(transpose(y) /= 0)) STOP 36
deallocate(foo,bar,x,y)
end
subroutine test_reshape
character(len=1) :: tempn(1,2)
character(len=1),allocatable :: foo(:,:), bar(:,:)
integer :: tempm(1,2)
integer,allocatable :: x(:,:), y(:,:)
tempn = 'b'
tempm = -42
allocate(foo(3,0),bar(-2:-4,7:9),x(3,0),y(-2:-4,7:9))
if (size(reshape(tempn(-7:-8,:),(/3,3/),pad=(/'a'/))) /= 9 .or. &
any(reshape(tempn(-7:-8,:),(/3,3/),pad=(/'a'/)) /= 'a')) STOP 37
if (size(reshape(tempn(-7:-8,:),(/3,3,3/),pad=(/'a'/))) /= 27 .or. &
any(reshape(tempn(-7:-8,:),(/3,3,3/),pad=(/'a'/)) /= 'a')) STOP 38
if (size(reshape(tempn(-7:-8,:),(/3,3,3,3,3,3,3/),pad=(/'a'/))) /= 2187 .or. &
any(reshape(tempn(-7:-8,:),(/3,3,3,3,3,3,3/),pad=(/'a'/)) /= 'a')) STOP 39
if (size(reshape(foo,(/3,3/),pad=(/'a'/))) /= 9 .or. &
any(reshape(foo,(/3,3/),pad=(/'a'/)) /= 'a')) STOP 40
if (size(reshape(foo,(/3,3,3/),pad=(/'a'/))) /= 27 .or. &
any(reshape(foo,(/3,3,3/),pad=(/'a'/)) /= 'a')) STOP 41
if (size(reshape(foo,(/3,3,3,3,3,3,3/),pad=(/'a'/))) /= 2187 .or. &
any(reshape(foo,(/3,3,3,3,3,3,3/),pad=(/'a'/)) /= 'a')) STOP 42
if (size(reshape(bar,(/3,3/),pad=(/'a'/))) /= 9 .or. &
any(reshape(bar,(/3,3/),pad=(/'a'/)) /= 'a')) STOP 43
if (size(reshape(bar,(/3,3,3/),pad=(/'a'/))) /= 27 .or. &
any(reshape(bar,(/3,3,3/),pad=(/'a'/)) /= 'a')) STOP 44
if (size(reshape(bar,(/3,3,3,3,3,3,3/),pad=(/'a'/))) /= 2187 .or. &
any(reshape(bar,(/3,3,3,3,3,3,3/),pad=(/'a'/)) /= 'a')) STOP 45
if (size(reshape(tempm(-7:-8,:),(/3,3/),pad=(/7/))) /= 9 .or. &
any(reshape(tempm(-7:-8,:),(/3,3/),pad=(/7/)) /= 7)) STOP 46
if (size(reshape(tempm(-7:-8,:),(/3,3,3/),pad=(/7/))) /= 27 .or. &
any(reshape(tempm(-7:-8,:),(/3,3,3/),pad=(/7/)) /= 7)) STOP 47
if (size(reshape(tempm(-7:-8,:),(/3,3,3,3,3,3,3/),pad=(/7/))) /= 2187 .or. &
any(reshape(tempm(-7:-8,:),(/3,3,3,3,3,3,3/),pad=(/7/)) /= 7)) STOP 48
if (size(reshape(x,(/3,3/),pad=(/7/))) /= 9 .or. &
any(reshape(x,(/3,3/),pad=(/7/)) /= 7)) STOP 49
if (size(reshape(x,(/3,3,3/),pad=(/7/))) /= 27 .or. &
any(reshape(x,(/3,3,3/),pad=(/7/)) /= 7)) STOP 50
if (size(reshape(x,(/3,3,3,3,3,3,3/),pad=(/7/))) /= 2187 .or. &
any(reshape(x,(/3,3,3,3,3,3,3/),pad=(/7/)) /= 7)) STOP 51
if (size(reshape(y,(/3,3/),pad=(/7/))) /= 9 .or. &
any(reshape(y,(/3,3/),pad=(/7/)) /= 7)) STOP 52
if (size(reshape(y,(/3,3,3/),pad=(/7/))) /= 27 .or. &
any(reshape(y,(/3,3,3/),pad=(/7/)) /= 7)) STOP 53
if (size(reshape(y,(/3,3,3,3,3,3,3/),pad=(/7/))) /= 2187 .or. &
any(reshape(y,(/3,3,3,3,3,3,3/),pad=(/7/)) /= 7)) STOP 54
deallocate(foo,bar,x,y)
end
subroutine test_pack
integer :: tempn(1,5)
integer,allocatable :: foo(:,:)
tempn = 2
allocate(foo(0,1:7))
if (size(pack(foo,foo/=0)) /= 0 .or. any(pack(foo,foo/=0) /= -42)) STOP 55
if (size(pack(foo,foo/=0,(/1,3,4,5,1,0,7,9/))) /= 8 .or. &
sum(pack(foo,foo/=0,(/1,3,4,5,1,0,7,9/))) /= 30) STOP 56
if (size(pack(tempn(:,-4:-5),tempn(:,-4:-5)/=0)) /= 0 .or. &
any(pack(tempn(:,-4:-5),tempn(:,-4:-5)/=0) /= -42)) STOP 57
if (size(pack(tempn(:,-4:-5),tempn(:,-4:-5)/=0,(/1,3,4,5,1,0,7,9/))) /= 8 .or. &
sum(pack(tempn(:,-4:-5),tempn(:,-4:-5)/=0,(/1,3,4,5,1,0,7,9/))) /= 30) &
STOP 58
if (size(pack(foo,.true.)) /= 0 .or. any(pack(foo,.true.) /= -42)) &
STOP 59
if (size(pack(foo,.true.,(/1,3,4,5,1,0,7,9/))) /= 8 .or. &
sum(pack(foo,.true.,(/1,3,4,5,1,0,7,9/))) /= 30) STOP 60
if (size(pack(tempn(:,-4:-5),.true.)) /= 0 .or. &
any(pack(foo,.true.) /= -42)) STOP 61
if (size(pack(tempn(:,-4:-5),.true.,(/1,3,4,5,1,0,7,9/))) /= 8 .or. &
sum(pack(tempn(:,-4:-5),.true.,(/1,3,4,5,1,0,7,9/))) /= 30) STOP 62
deallocate(foo)
end
subroutine test_unpack
integer :: tempn(1,5), tempv(5)
integer,allocatable :: foo(:,:), bar(:)
integer :: zero
tempn = 2
tempv = 5
zero = 0
allocate(foo(0,1:7),bar(0:-1))
if (any(unpack(tempv,tempv/=0,tempv) /= 5) .or. &
size(unpack(tempv,tempv/=0,tempv)) /= 5) STOP 63
if (any(unpack(tempv(1:0),tempv/=0,tempv) /= 5) .or. &
size(unpack(tempv(1:0),tempv/=0,tempv)) /= 5) STOP 64
if (any(unpack(tempv,tempv(1:zero)/=0,tempv) /= -47)) STOP 65
if (any(unpack(tempv(5:4),tempv(1:zero)/=0,tempv) /= -47)) STOP 66
if (any(unpack(bar,foo==foo,foo) /= -47)) STOP 67
deallocate(foo,bar)
end
subroutine test_spread
real :: tempn(1)
real,allocatable :: foo(:)
tempn = 2.0
allocate(foo(0))
if (any(spread(1,dim=1,ncopies=0) /= -17.0) .or. &
size(spread(1,dim=1,ncopies=0)) /= 0) STOP 68
if (any(spread(foo,dim=1,ncopies=1) /= -17.0) .or. &
size(spread(foo,dim=1,ncopies=1)) /= 0) STOP 69
if (any(spread(tempn(2:1),dim=1,ncopies=1) /= -17.0) .or. &
size(spread(tempn(2:1),dim=1,ncopies=1)) /= 0) STOP 70
deallocate(foo)
end
program test
call test_cshift
call test_eoshift
call test_transpose
call test_unpack
call test_spread
call test_pack
call test_reshape
end
|
