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
|
function gbtest122
%GBTEST122 test reshape (extended methods in GrB)
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
rng ('default')
for m = 1:12
fprintf ('.') ;
for n = 1:12
for kind = [0 1]
if (kind == 0)
A = rand (m, n) ;
else
A = sprand (m, n, 0.3) ;
end
G = GrB (A) ;
mn = m*n ;
H = GrB (A, 'by row') ;
f = factor (mn) ;
for k = 1:length (f)
S = nchoosek (f, k) ;
for i = 1:size(S,1)
% reshape by column
m2 = prod (S (i,:)) ;
n2 = mn / m2 ;
C1 = reshape (A, m2, n2) ;
C2 = reshape (G, m2, n2) ;
assert (gbtest_eq (C1, C2)) ;
C3 = reshape (H, m2, n2) ;
assert (gbtest_eq (C1, C3)) ;
C1 = reshape (A, [m2 n2]) ;
C2 = reshape (G, [m2 n2]) ;
assert (gbtest_eq (C1, C2)) ;
C3 = reshape (H, [m2 n2]) ;
assert (gbtest_eq (C1, C3)) ;
% reshape by row
C1 = reshape (A', n2, m2)' ;
C2 = reshape (G, m2, n2, 'by row') ;
assert (gbtest_eq (C1, C2)) ;
C3 = reshape (H, m2, n2, 'by row') ;
assert (gbtest_eq (C1, C3)) ;
C2 = reshape (G, [m2 n2], 'by row') ;
assert (gbtest_eq (C1, C2)) ;
C3 = reshape (H, [m2 n2], 'by row') ;
assert (gbtest_eq (C1, C3)) ;
end
end
end
end
end
fprintf ('\ngbtest122: all tests passed\n') ;
|