1
- <<<<<< < HEAD
2
1
using Test, LinearAlgebra, SpecialFunctions
3
- ====== =
4
- using SpecialFunctions
5
- using LinearAlgebra
6
- >>>>>> > scale! -> [lr]mul!
2
+ using Statistics: mean
3
+ @everywhere using SparseArrays: sprandn
7
4
8
5
@testset " test distribute and other constructors" begin
9
6
A = rand (1 : 100 , (100 ,100 ))
@@ -210,29 +207,17 @@ end
210
207
211
208
check_leaks ()
212
209
213
- <<<<<< < HEAD
214
210
@testset " test rmul!(Diagonal, A)" begin
215
- ====== =
216
- @testset " test [lr]mul!(b, A)" begin
217
- >>>>>> > scale! -> [lr]mul!
218
211
A = randn (100 , 100 )
219
212
b = randn (100 )
220
213
D = Diagonal (b)
221
214
DA = distribute (A)
222
- <<<<<< < HEAD
223
215
@test lmul! (D, A) == lmul! (D, DA)
224
- ====== =
225
- @test lmul! (Diagonal (b), A) == lmul! (Diagonal (b), DA)
226
- >>>>>> > scale! -> [lr]mul!
227
216
close (DA)
228
217
A = randn (100 , 100 )
229
218
b = randn (100 )
230
219
DA = distribute (A)
231
- <<<<<< < HEAD
232
220
@test rmul! (A, D) == rmul! (DA, D)
233
- ====== =
234
- @test rmul! (A, Diagonal (b)) == rmul! (A, Diagonal (b))
235
- >>>>>> > scale! -> [lr]mul!
236
221
close (DA)
237
222
end
238
223
@@ -242,6 +227,7 @@ check_leaks()
242
227
for _ = 1 : 25 , f = [x -> Int128 (2 x), x -> Int128 (x^ 2 ), x -> Int128 (x^ 2 + 2 x - 1 )], opt = [+ , * ]
243
228
A = rand (1 : 5 , rand (2 : 30 ))
244
229
DA = distribute (A)
230
+ @test DA isa DArray
245
231
@test mapreduce (f, opt, DA) - mapreduce (f, opt, A) == 0
246
232
close (DA)
247
233
end
@@ -252,15 +238,16 @@ check_leaks()
252
238
@testset " test mapreducedim on DArrays" begin
253
239
D = DArray (I-> fill (myid (), map (length,I)), (73 ,73 ), [MYID, OTHERIDS])
254
240
D2 = map (x-> 1 , D)
255
- @test mapreducedim (t -> t* t, + , D2, 1 ) == mapreducedim (t -> t* t, + , convert (Array, D2), 1 )
256
- @test mapreducedim (t -> t* t, + , D2, 2 ) == mapreducedim (t -> t* t, + , convert (Array, D2), 2 )
257
- @test mapreducedim (t -> t* t, + , D2, (1 ,2 )) == mapreducedim (t -> t* t, + , convert (Array, D2), (1 ,2 ))
241
+ @test D2 isa DArray
242
+ @test mapreduce (t -> t* t, + , D2, dims= 1 ) == mapreduce (t -> t* t, + , convert (Array, D2), dims= 1 )
243
+ @test mapreduce (t -> t* t, + , D2, dims= 2 ) == mapreduce (t -> t* t, + , convert (Array, D2), dims= 2 )
244
+ @test mapreduce (t -> t* t, + , D2, dims= (1 ,2 )) == mapreduce (t -> t* t, + , convert (Array, D2), dims= (1 ,2 ))
258
245
259
246
# Test non-regularly chunked DArrays
260
247
r1 = DistributedArrays. remotecall (() -> sprandn (3 , 10 , 0.1 ), workers ()[1 ])
261
248
r2 = DistributedArrays. remotecall (() -> sprandn (7 , 10 , 0.1 ), workers ()[2 ])
262
249
D = DArray (reshape ([r1; r2], (2 ,1 )))
263
- @test Array (sum (D, 2 )) == sum (Array (D), 2 )
250
+ @test Array (sum (D, dims = 2 )) == sum (Array (D), dims = 2 )
264
251
265
252
# close(D)
266
253
# close(D2)
@@ -275,10 +262,10 @@ check_leaks()
275
262
A = convert (Array, DA)
276
263
277
264
@testset " dimension $dms " for dms in (1 , 2 , 3 , (1 ,2 ), (1 ,3 ), (2 ,3 ), (1 ,2 ,3 ))
278
- @test mapreducedim (t -> t* t, + , A, dms) ≈ mapreducedim (t -> t* t, + , DA, dms)
279
- @test mapreducedim (t -> t* t, + , A, dms, 1.0 ) ≈ mapreducedim (t -> t* t, + , DA, dms, 1.0 )
280
- @test reducedim (* , A, dms) ≈ reducedim (* , DA, dms)
281
- @test reducedim (* , A, dms, 2.0 ) ≈ reducedim (* , DA, dms, 2.0 )
265
+ @test mapreduce (t -> t* t, + , A, dims = dms) ≈ mapreduce (t -> t* t, + , DA, dims = dms)
266
+ @test mapreduce (t -> t* t, + , A, dims = dms, init = 1.0 ) ≈ mapreduce (t -> t* t, + , DA, dims = dms, init = 1.0 )
267
+ @test reduce (* , A, dims = dms) ≈ reduce (* , DA, dims = dms)
268
+ @test reduce (* , A, dims = dms, init = 2.0 ) ≈ reduce (* , DA, dims = dms, init = 2.0 )
282
269
end
283
270
close (DA)
284
271
d_closeall () # temp created by the mapreduce above
@@ -293,7 +280,7 @@ check_leaks()
293
280
294
281
@testset " test $f for dimension $dms " for f in (mean, ), dms in (1 , 2 , 3 , (1 ,2 ), (1 ,3 ), (2 ,3 ), (1 ,2 ,3 ))
295
282
# std is pending implementation
296
- @test f (DA,dms) ≈ f (A,dms)
283
+ @test f (DA, dims = dms) ≈ f (A, dims = dms)
297
284
end
298
285
299
286
close (DA)
@@ -308,7 +295,7 @@ check_leaks()
308
295
309
296
# sum either throws an ArgumentError or a CompositeException of ArgumentErrors
310
297
try
311
- sum (DA, - 1 )
298
+ sum (DA, dims = - 1 )
312
299
catch err
313
300
if isa (err, CompositeException)
314
301
@test ! isempty (err. exceptions)
@@ -322,7 +309,7 @@ check_leaks()
322
309
end
323
310
end
324
311
try
325
- sum (DA, 0 )
312
+ sum (DA, dims = 0 )
326
313
catch err
327
314
if isa (err, CompositeException)
328
315
@test ! isempty (err. exceptions)
@@ -337,9 +324,9 @@ check_leaks()
337
324
end
338
325
339
326
@test sum (DA) ≈ sum (A)
340
- @test sum (DA,1 ) ≈ sum (A,1 )
341
- @test sum (DA,2 ) ≈ sum (A,2 )
342
- @test sum (DA,3 ) ≈ sum (A,3 )
327
+ @test sum (DA, dims = 1 ) ≈ sum (A, dims = 1 )
328
+ @test sum (DA, dims = 2 ) ≈ sum (A, dims = 2 )
329
+ @test sum (DA, dims = 3 ) ≈ sum (A, dims = 3 )
343
330
close (DA)
344
331
d_closeall () # temporaries created above
345
332
end
360
347
361
348
check_leaks ()
362
349
363
- # test length / endof
350
+ # test length / lastindex
364
351
@testset " test collections API" begin
365
352
A = randn (23 ,23 )
366
353
DA = distribute (A)
@@ -369,16 +356,16 @@ check_leaks()
369
356
@test length (DA) == length (A)
370
357
end
371
358
372
- @testset " test endof " begin
373
- @test endof (DA) == endof (A)
359
+ @testset " test lastindex " begin
360
+ @test lastindex (DA) == lastindex (A)
374
361
end
375
362
close (DA)
376
363
end
377
364
378
365
check_leaks ()
379
366
380
367
@testset " test max / min / sum" begin
381
- a = map (x -> Int (round (rand () * 100 )) - 50 , Array {Int} (100 ,1000 ))
368
+ a = map (x -> Int (round (rand () * 100 )) - 50 , Array {Int} (undef, 100 ,1000 ))
382
369
d = distribute (a)
383
370
384
371
@test sum (d) == sum (a)
395
382
check_leaks ()
396
383
397
384
@testset " test all / any" begin
398
- a = map (x-> Int (round (rand () * 100 )) - 50 , Array {Int} (100 ,1000 ))
385
+ a = map (x-> Int (round (rand () * 100 )) - 50 , Array {Int} (undef, 100 ,1000 ))
399
386
a = [true for i in 1 : 100 ]
400
387
d = distribute (a)
401
388
@@ -651,25 +638,25 @@ end
651
638
652
639
check_leaks ()
653
640
654
- @testset " test c/ transpose" begin
655
- @testset " test ctranspose real" begin
641
+ @testset " test transpose/adjoint " begin
642
+ @testset " test transpose real" begin
656
643
A = drand (Float64, 100 , 200 )
657
- @test A ' == Array (A)'
644
+ @test transpose (A) == transpose ( Array (A))
658
645
close (A)
659
646
end
660
- @testset " test ctranspose complex" begin
661
- A = drand (Complex128 , 200 , 100 )
662
- @test A ' == Array (A)'
647
+ @testset " test transpose complex" begin
648
+ A = drand (ComplexF64 , 200 , 100 )
649
+ @test transpose (A) == transpose ( Array (A))
663
650
close (A)
664
651
end
665
- @testset " test transpose real" begin
652
+ @testset " test adjoint real" begin
666
653
A = drand (Float64, 200 , 100 )
667
- @test transpose (A) == transpose (Array (A))
654
+ @test adjoint (A) == adjoint (Array (A))
668
655
close (A)
669
656
end
670
- @testset " test ctranspose complex" begin
671
- A = drand (Complex128 , 100 , 200 )
672
- @test transpose (A) == transpose (Array (A))
657
+ @testset " test adjoint complex" begin
658
+ A = drand (ComplexF64 , 100 , 200 )
659
+ @test adjoint (A) == adjoint (Array (A))
673
660
close (A)
674
661
end
675
662
@@ -712,8 +699,8 @@ check_leaks()
712
699
sinh, sinpi, sqrt, tan, tand, tanh, trigamma)
713
700
@test f .(a) == f .(b)
714
701
end
715
- a = a + 1
716
- b = b + 1
702
+ a = a . + 1
703
+ b = b . + 1
717
704
@testset " $f " for f in (asec, asecd, acosh, acsc, acscd, acoth)
718
705
@test f .(a) == f .(b)
719
706
end
0 commit comments