diff test/logical-index.tst @ 19010:3fb030666878 draft default tip dspies

Added special-case logical-indexing function * logical-index.h (New file) : Logical-indexing function. May be called on octave_value types via call_bool_index * nz-iterators.h : Add base-class nz_iterator for iterator types. Array has template bool for whether to internally store row-col or compute on the fly Add skip_ahead method which skips forward to the next nonzero after its argument Add flat_index for computing octave_idx_type index of current position (with assertion failure in the case of overflow) Move is_zero to separate file * ov-base-diag.cc, ov-base-mat.cc, ov-base-sparse.cc, ov-perm.cc (do_index_op): Add call to call_bool_index in logical-index.h * Array.h : Move forward-declaration for array_iterator to separate header file * dim-vector.cc (dim_max): Refers to idx-bounds.h (max_idx) * array-iter-decl.h (New file): Header file for forward declaration of array-iterator * direction.h : Add constants fdirc and bdirc to avoid having to reconstruct them * dv-utils.h, dv-utils.cc (New files) : Utility functions for querying and constructing dim-vectors * idx-bounds.h (New file) : Utility constants and functions for determining whether things will overflow the maximum allowed bounds * interp-idx.h (New function : to_flat_idx) : Converts row-col pair to linear index of octave_idx_type * is-zero.h (New file) : Function for determining whether an element is zero * logical-index.tst : Add tests for correct return-value dimensions and large sparse matrix behavior
author David Spies <dnspies@gmail.com>
date Fri, 25 Jul 2014 13:39:31 -0600
parents d63878346099
children
line wrap: on
line diff
--- a/test/logical-index.tst	Mon Jul 14 13:07:59 2014 -0600
+++ b/test/logical-index.tst	Fri Jul 25 13:39:31 2014 -0600
@@ -30,13 +30,18 @@
 %!assert (isempty (a(logical ([0,0,0,0]))))
 %!assert (a(logical ([1,1,1,1])), [9,8,7,6])
 %!assert (a(logical ([0,1,1,0])), [8,7])
+%!assert (a(logical ([0;1;1;0])), [8,7])
 %!assert (a(logical ([1,1])), [9,8])
 
+%! a = permute(a,[1,3,2]);
+%!assert (a(logical ([0,1,1,0])), permute([8,7],[1,3,2]))
+
 %!shared a
 %! a = [9,8;7,6];
-%!assert (isempty (a(logical ([0,0,0,0]))))
+%!assert (a(logical ([0,0,0,0])), zeros(1,0))
 %!assert (a(logical ([1,1,1,1])), [9,7,8,6])
 %!assert (a(logical ([0,1,1,0])), [7,8])
+%!assert (a(logical ([0;1;1;0])), [7;8])
 %!assert (a(logical (0:1),logical (0:1)), 6)
 %!assert (a(logical (0:1),2:-1:1), [6,7])
 %!assert (a(logical (0:1),logical ([0,1])), 6)
@@ -71,3 +76,21 @@
 %!assert (a(logical ([1,1]),1), [9;7])
 %!assert (a(logical ([1,1]),logical ([1,1])), [9,8;7,6])
 
+%!assert (a(logical (permute([1,1,1,1],[1,3,2]))), [9;7;8;6])
+%!assert (a(logical (permute([0,1,1,0],[1,3,2]))), [7;8])
+%!assert (a(logical([])), [])
+%!assert (a(false), [])
+%!assert (a(false(0,0,1)), [])
+%!assert (a(false(2,2)),zeros(0,1))
+%!assert (a(false(1,4)),zeros(1,0))
+%!assert (a(false(4,1)),zeros(0,1))
+%!assert (a(false(1,1,4)),zeros(0,1))
+
+%!shared v,a,b,c
+%! v = sparse((1:100000).');
+%! v = v .* mod(v,2);
+%! a = sparse(diag(v));
+%! b = logical(speye(100000));
+%! c = [b(:,1:2:end); b(:,2:2:end)];
+%!assert (a(b), v)
+%!assert (a(c), v)