Fix empty array bug in array-index-map!
[bpt/guile.git] / test-suite / tests / ramap.test
1 ;;;; ramap.test --- test array mapping functions -*- scheme -*-
2 ;;;;
3 ;;;; Copyright (C) 2004, 2005, 2006, 2009, 2013 Free Software Foundation, Inc.
4 ;;;;
5 ;;;; This library is free software; you can redistribute it and/or
6 ;;;; modify it under the terms of the GNU Lesser General Public
7 ;;;; License as published by the Free Software Foundation; either
8 ;;;; version 3 of the License, or (at your option) any later version.
9 ;;;;
10 ;;;; This library is distributed in the hope that it will be useful,
11 ;;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
12 ;;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 ;;;; Lesser General Public License for more details.
14 ;;;;
15 ;;;; You should have received a copy of the GNU Lesser General Public
16 ;;;; License along with this library; if not, write to the Free Software
17 ;;;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
19 (define-module (test-suite test-ramap)
20 #:use-module (test-suite lib))
21
22 (define (array-row a i)
23 (make-shared-array a (lambda (j) (list i j))
24 (cadr (array-dimensions a))))
25
26 (define (array-col a j)
27 (make-shared-array a (lambda (i) (list i j))
28 (car (array-dimensions a))))
29
30 ;;;
31 ;;; array-index-map!
32 ;;;
33
34 (with-test-prefix "array-index-map!"
35
36 (pass-if "basic test"
37 (let ((nlst '()))
38 (array-index-map! (make-array #f '(1 1))
39 (lambda (n)
40 (set! nlst (cons n nlst))))
41 (equal? nlst '(1))))
42
43 (with-test-prefix "empty arrays"
44
45 (pass-if "all axes empty"
46 (array-index-map! (make-typed-array 'f64 0 0 0) (const 0))
47 (array-index-map! (make-typed-array 'b #t 0 0) (const #t))
48 (array-index-map! (make-typed-array #t 0 0 0) (const 0)))
49
50 (pass-if "last axis empty"
51 (array-index-map! (make-typed-array 'f64 0 2 0) (const 0))
52 (array-index-map! (make-typed-array 'b #t 2 0) (const #t))
53 (array-index-map! (make-typed-array #t 0 2 0) (const 0)))
54
55 (pass-if "axis empty, other than last"
56 (array-index-map! (make-typed-array 'f64 0 0 2) (const 0))
57 (array-index-map! (make-typed-array 'b #t 0 2) (const #t))
58 (array-index-map! (make-typed-array #t 0 0 2) (const 0)))))
59
60 ;;;
61 ;;; array-copy!
62 ;;;
63
64 (with-test-prefix "array-copy!"
65
66 (pass-if "empty arrays"
67 (let* ((b (make-array 0 2 2))
68 (c (make-shared-array b (lambda (i j) (list i j)) 0 2)))
69 (array-copy! #2:0:2() c)
70 (array-equal? #2:0:2() c))))
71
72 ;;;
73 ;;; array-map!
74 ;;;
75
76 (with-test-prefix "array-map!"
77
78 (pass-if-exception "no args" exception:wrong-num-args
79 (array-map!))
80
81 (pass-if-exception "one arg" exception:wrong-num-args
82 (array-map! (make-array #f 5)))
83
84 (with-test-prefix "no sources"
85
86 (pass-if "closure 0"
87 (array-map! (make-array #f 5) (lambda () #f))
88 #t)
89
90 (pass-if-exception "closure 1" exception:wrong-num-args
91 (array-map! (make-array #f 5) (lambda (x) #f)))
92
93 (pass-if-exception "closure 2" exception:wrong-num-args
94 (array-map! (make-array #f 5) (lambda (x y) #f)))
95
96 (pass-if-exception "subr_1" exception:wrong-num-args
97 (array-map! (make-array #f 5) length))
98
99 (pass-if-exception "subr_2" exception:wrong-num-args
100 (array-map! (make-array #f 5) logtest))
101
102 (pass-if-exception "subr_2o" exception:wrong-num-args
103 (array-map! (make-array #f 5) number->string))
104
105 (pass-if-exception "dsubr" exception:wrong-num-args
106 (array-map! (make-array #f 5) sqrt))
107
108 (pass-if "rpsubr"
109 (let ((a (make-array 'foo 5)))
110 (array-map! a =)
111 (equal? a (make-array #t 5))))
112
113 (pass-if "asubr"
114 (let ((a (make-array 'foo 5)))
115 (array-map! a +)
116 (equal? a (make-array 0 5))))
117
118 ;; in Guile 1.6.4 and earlier this resulted in a segv
119 (pass-if "noop"
120 (array-map! (make-array #f 5) noop)
121 #t))
122
123 (with-test-prefix "one source"
124
125 (pass-if-exception "closure 0" exception:wrong-num-args
126 (array-map! (make-array #f 5) (lambda () #f)
127 (make-array #f 5)))
128
129 (pass-if "closure 1"
130 (let ((a (make-array #f 5)))
131 (array-map! a (lambda (x) 'foo) (make-array #f 5))
132 (equal? a (make-array 'foo 5))))
133
134 (pass-if-exception "closure 2" exception:wrong-num-args
135 (array-map! (make-array #f 5) (lambda (x y) #f)
136 (make-array #f 5)))
137
138 (pass-if "subr_1"
139 (let ((a (make-array #f 5)))
140 (array-map! a length (make-array '(x y z) 5))
141 (equal? a (make-array 3 5))))
142
143 (pass-if-exception "subr_2" exception:wrong-num-args
144 (array-map! (make-array #f 5) logtest
145 (make-array 999 5)))
146
147 (pass-if "subr_2o"
148 (let ((a (make-array #f 5)))
149 (array-map! a number->string (make-array 99 5))
150 (equal? a (make-array "99" 5))))
151
152 (pass-if "dsubr"
153 (let ((a (make-array #f 5)))
154 (array-map! a sqrt (make-array 16.0 5))
155 (equal? a (make-array 4.0 5))))
156
157 (pass-if "rpsubr"
158 (let ((a (make-array 'foo 5)))
159 (array-map! a = (make-array 0 5))
160 (equal? a (make-array #t 5))))
161
162 (pass-if "asubr"
163 (let ((a (make-array 'foo 5)))
164 (array-map! a - (make-array 99 5))
165 (equal? a (make-array -99 5))))
166
167 ;; in Guile 1.6.5 and 1.6.6 this was an error
168 (pass-if "1+"
169 (let ((a (make-array #f 5)))
170 (array-map! a 1+ (make-array 123 5))
171 (equal? a (make-array 124 5)))))
172
173 (with-test-prefix "two sources"
174
175 (pass-if-exception "closure 0" exception:wrong-num-args
176 (array-map! (make-array #f 5) (lambda () #f)
177 (make-array #f 5) (make-array #f 5)))
178
179 (pass-if-exception "closure 1" exception:wrong-num-args
180 (array-map! (make-array #f 5) (lambda (x) #f)
181 (make-array #f 5) (make-array #f 5)))
182
183 (pass-if "closure 2"
184 (let ((a (make-array #f 5)))
185 (array-map! a (lambda (x y) 'foo)
186 (make-array #f 5) (make-array #f 5))
187 (equal? a (make-array 'foo 5))))
188
189 (pass-if-exception "subr_1" exception:wrong-num-args
190 (array-map! (make-array #f 5) length
191 (make-array #f 5) (make-array #f 5)))
192
193 (pass-if "subr_2"
194 (let ((a (make-array 'foo 5)))
195 (array-map! a logtest
196 (make-array 999 5) (make-array 999 5))
197 (equal? a (make-array #t 5))))
198
199 (pass-if "subr_2o"
200 (let ((a (make-array #f 5)))
201 (array-map! a number->string
202 (make-array 32 5) (make-array 16 5))
203 (equal? a (make-array "20" 5))))
204
205 (pass-if-exception "dsubr" exception:wrong-num-args
206 (let ((a (make-array #f 5)))
207 (array-map! a sqrt
208 (make-array 16.0 5) (make-array 16.0 5))
209 (equal? a (make-array 4.0 5))))
210
211 (pass-if "rpsubr"
212 (let ((a (make-array 'foo 5)))
213 (array-map! a = (make-array 99 5) (make-array 77 5))
214 (equal? a (make-array #f 5))))
215
216 (pass-if "asubr"
217 (let ((a (make-array 'foo 5)))
218 (array-map! a - (make-array 99 5) (make-array 11 5))
219 (equal? a (make-array 88 5))))
220
221 (pass-if "+"
222 (let ((a (make-array #f 4)))
223 (array-map! a + #(1 2 3 4) #(5 6 7 8))
224 (equal? a #(6 8 10 12))))
225
226 (pass-if "noncompact arrays 1"
227 (let ((a #2((0 1) (2 3)))
228 (c (make-array 0 2)))
229 (begin
230 (array-map! c + (array-row a 1) (array-row a 1))
231 (array-equal? c #(4 6)))))
232
233 (pass-if "noncompact arrays 2"
234 (let ((a #2((0 1) (2 3)))
235 (c (make-array 0 2)))
236 (begin
237 (array-map! c + (array-col a 1) (array-col a 1))
238 (array-equal? c #(2 6)))))
239
240 (pass-if "noncompact arrays 3"
241 (let ((a #2((0 1) (2 3)))
242 (c (make-array 0 2)))
243 (begin
244 (array-map! c + (array-col a 1) (array-row a 1))
245 (array-equal? c #(3 6)))))
246
247 (pass-if "noncompact arrays 4"
248 (let ((a #2((0 1) (2 3)))
249 (c (make-array 0 2)))
250 (begin
251 (array-map! c + (array-col a 1) (array-row a 1))
252 (array-equal? c #(3 6)))))))
253
254 ;;;
255 ;;; array-for-each
256 ;;;
257
258 (with-test-prefix "array-for-each"
259
260 (with-test-prefix "1 source"
261 (pass-if-equal "noncompact array"
262 '(3 2 1 0)
263 (let* ((a #2((0 1) (2 3)))
264 (l '())
265 (p (lambda (x) (set! l (cons x l)))))
266 (array-for-each p a)
267 l))
268
269 (pass-if-equal "vector"
270 '(3 2 1 0)
271 (let* ((a #(0 1 2 3))
272 (l '())
273 (p (lambda (x) (set! l (cons x l)))))
274 (array-for-each p a)
275 l))
276
277 (pass-if-equal "shared array"
278 '(3 2 1 0)
279 (let* ((a #2((0 1) (2 3)))
280 (a' (make-shared-array a
281 (lambda (x)
282 (list (quotient x 4)
283 (modulo x 4)))
284 4))
285 (l '())
286 (p (lambda (x) (set! l (cons x l)))))
287 (array-for-each p a')
288 l)))
289
290 (with-test-prefix "3 sources"
291 (pass-if-equal "noncompact arrays 1"
292 '((3 3 3) (2 2 2))
293 (let* ((a #2((0 1) (2 3)))
294 (l '())
295 (rec (lambda args (set! l (cons args l)))))
296 (array-for-each rec (array-row a 1) (array-row a 1) (array-row a 1))
297 l))
298
299 (pass-if-equal "noncompact arrays 2"
300 '((3 3 3) (2 2 1))
301 (let* ((a #2((0 1) (2 3)))
302 (l '())
303 (rec (lambda args (set! l (cons args l)))))
304 (array-for-each rec (array-row a 1) (array-row a 1) (array-col a 1))
305 l))
306
307 (pass-if-equal "noncompact arrays 3"
308 '((3 3 3) (2 1 1))
309 (let* ((a #2((0 1) (2 3)))
310 (l '())
311 (rec (lambda args (set! l (cons args l)))))
312 (array-for-each rec (array-row a 1) (array-col a 1) (array-col a 1))
313 l))
314
315 (pass-if-equal "noncompact arrays 4"
316 '((3 2 3) (1 0 2))
317 (let* ((a #2((0 1) (2 3)))
318 (l '())
319 (rec (lambda args (set! l (cons args l)))))
320 (array-for-each rec (array-col a 1) (array-col a 0) (array-row a 1))
321 l))))