Fix bug. Add test for subindex.

This commit is contained in:
Adrian Mariano 2020-08-02 10:38:11 -04:00
parent 37ca6efebb
commit 06b69475cb
2 changed files with 2 additions and 1 deletions

View file

@ -1187,7 +1187,7 @@ function add_scalar(v,s) =
// subindex(M,[2]); // Returns [[3], [7], [11], [15]]
// subindex(M,[2,1]); // Returns [[3, 2], [7, 6], [11, 10], [15, 14]]
// subindex(M,[1:3]); // Returns [[2, 3, 4], [6, 7, 8], [10, 11, 12], [14, 15, 16]]
function subindex(v, idx) =
function subindex(M, idx) =
is_num(idx)
? [for(row=M) row[idx]]
: [for(row=M) [for(i=idx) row[i]]];

View file

@ -358,6 +358,7 @@ test_add_scalar();
module test_subindex() {
v = [[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]];
assert(subindex(v,2) == [3, 7, 11, 15]);
assert(subindex(v,[2]) == [[3], [7], [11], [15]]);
assert(subindex(v,[2,1]) == [[3, 2], [7, 6], [11, 10], [15, 14]]);
assert(subindex(v,[1:3]) == [[2, 3, 4], [6, 7, 8], [10, 11, 12], [14, 15, 16]]);
}