Refactor inclusion of system.i to one place
[Math-GSL.git] / t / Vector.t
blobfcbf47b04e5a0326f6f13d7814b90beb326bc541
1 package Math::GSL::Vector::Test;
2 use base q{Test::Class};
3 use Test::More tests => 136;
4 use Math::GSL          qw/:all/;
5 use Math::GSL::Test    qw/:all/;
6 use Math::GSL::Errno   qw/:all/;
7 use Math::GSL::Vector  qw/:all/;
8 use Test::Exception;
9 use Data::Dumper;
10 use strict;
12 BEGIN{ gsl_set_error_handler_off(); }
14 sub make_fixture : Test(setup) {
15     my $self = shift;
16     $self->{vector} = gsl_vector_alloc(5);
17     $self->{object} = Math::GSL::Vector->new([1 .. 5 ]);
20 sub teardown : Test(teardown) {
21     unlink 'vector' if -f 'vector';
24 sub GSL_VECTOR_RAW : Tests {
25     my $vec = Math::GSL::Vector->new(10);
26     isa_ok($vec->raw, 'Math::GSL::Vector::gsl_vector');
29 sub GSL_VECTOR_ALLOC : Tests {
30     my $vector = gsl_vector_alloc(5);
31     isa_ok($vector, 'Math::GSL::Vector');
33 sub GSL_VECTOR_LENGTH: Tests {
34     my $self = shift;
35     my $vector = $self->{object};
36     ok( $vector->length == 5, '$vector->length' );
38 sub GSL_VECTOR_SET_GET: Tests { 
39     my $self = shift;
40     gsl_vector_set($self->{vector}, 0, 42 );
41     my $elem   = gsl_vector_get($self->{vector}, 0);
42     ok( $elem == 42, 'gsl_vector_set/gsl_vector_get' );
45 sub GSL_VECTOR_ISNONNEG: Tests {
46     my $self = shift;
47     map { gsl_vector_set($self->{vector}, $_, -1 ) } (0..4); 
48     ok( !gsl_vector_isnonneg($self->{vector}),'gsl_vector_isnonneg' );
49     map { gsl_vector_set($self->{vector}, $_, 1 ) } (0..4); 
50     ok( gsl_vector_isnonneg($self->{vector}),'gsl_vector_isnonneg' );
53 sub GSL_VECTOR_ISNULL: Tests(2) {
54     my $self = shift;
55     map { gsl_vector_set($self->{vector}, $_, 0 ) } (0..4); 
56     ok( gsl_vector_isnull($self->{vector}),'null vector returns true' );
57     gsl_vector_set($self->{vector}, 0, 5 );
58     ok( !gsl_vector_isnull($self->{vector}), 'changed non-null vector returns false' );
61 sub GSL_VECTOR_ISPOS: Tests {
62     my $self = shift;
63     map { gsl_vector_set($self->{vector}, $_, -1 ) } (0..4); 
64     ok( !gsl_vector_ispos($self->{vector}),'gsl_vector_pos' );
65     map { gsl_vector_set($self->{vector}, $_, 1 ) } (0..4); 
66     ok( gsl_vector_ispos($self->{vector}),'gsl_vector_pos' );
69 sub GSL_VECTOR_ISNEG: Tests {
70     my $self = shift;
72     map { gsl_vector_set($self->{vector}, $_, -$_ ) } (0..4); 
73     ok( !gsl_vector_isneg($self->{vector}),'gsl_vector_neg' );
75     gsl_vector_set($self->{vector}, 0, -1 );
77     ok( gsl_vector_isneg($self->{vector}),'gsl_vector_neg' );
80 sub GSL_VECTOR_NEW: Tests {
81     my $vec = Math::GSL::Vector->new( [ map { $_ ** 2 } (1..10) ] );
82     isa_ok( $vec, 'Math::GSL::Vector', 'Math::GSL::Vector->new($values)' );
84     dies_ok( sub { Math::GSL::Vector->new(-1) }, 'new takes only positive indices');
86     dies_ok( sub { Math::GSL::Vector->new(3.14) },  'new takes only integer indices');
88     dies_ok( sub { Math::GSL::Vector->new([]) },'new takes only nonempty array refs');
90     $vec = Math::GSL::Vector->new(42);
91     ok( $vec->length == 42 , 'new creates empty vectors of a given length');
93 sub GSL_VECTOR_AS_LIST: Tests {
94     my $vec = Math::GSL::Vector->new( [ map { $_ ** 2 } (reverse 1..10) ] );
95     my @x = $vec->as_list;
96     is_deeply( \@x, [map { $_ ** 2 } (reverse 1..10)] );
99 sub GSL_VECTOR_SET: Tests {
100     my $vec = Math::GSL::Vector->new( [ map { $_ ** 2 } (1..10) ] );
101     $vec->set( [ 0..4] , [ reverse 1..5 ] );
102     my ($x) = $vec->get([0]);
103     ok( $x == 5, "gsl_vector_set: $x ?= 5" );
105 sub GSL_VECTOR_MIN: Tests {
106     my $vec = Math::GSL::Vector->new( [ map { $_ ** 2 } (0..4) ] );
107     ok_similar( $vec->min ,0, '$vec->min' );
108     ok_similar( gsl_vector_min($vec->raw) ,0, 'gsl_vector_min' );
111 sub GSL_VECTOR_MAX: Tests {
112     my $vec = Math::GSL::Vector->new( [ 3, 567, 4200 ]);
113     ok_similar( $vec->max ,4200, '$vec->min' );
114     ok_similar( gsl_vector_max($vec->raw) ,4200, 'gsl_vector_max' );
116 sub GSL_VECTOR_FREAD_FWRITE: Tests { 
117     my $self = shift;
118     map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
120     my $fh = gsl_fopen("vector", 'w');
121     my $status = gsl_vector_fwrite($fh, $self->{vector} );
122     ok( ! $status, 'gsl_vector_fwrite' );
123     ok( -f "vector", 'gsl_vector_fwrite' );
124     ok_status(fclose($fh));
126     map { gsl_vector_set($self->{vector}, $_, $_ ** 3 ) } (0..4); 
128     $fh = gsl_fopen("vector", 'r');
130     ok_status(gsl_vector_fread($fh, $self->{vector} ));
131     is_deeply( [ map { gsl_vector_get($self->{vector}, $_) } (0..4) ],
132                [ map { $_ ** 2 } (0..4) ],
133              );
134     ok_status(gsl_fclose($fh));
137 sub GSL_VECTOR_SUBVECTOR : Tests {
138     my $self = shift;
139     map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
140     my $vec_sub = gsl_vector_subvector($self->{vector}, 2, 3);
142     ok_similar( [ map { gsl_vector_get($vec_sub->{vector}, $_) } (0..2) ], 
143                 [ 4, 9, 16 ], 
144               );
147 sub GSL_VECTOR_CALLOC : Tests {
148    my $vector = gsl_vector_calloc(5);
149    isa_ok($vector, 'Math::GSL::Vector');
152 sub GSL_VECTOR_SET_ALL : Tests {
153    my $vec = Math::GSL::Vector->new(5);
154    gsl_vector_set_all($vec->raw, 4);
155    ok_similar( [ $vec->as_list ], [ (4) x 5 ] );
158 sub GSL_VECTOR_SET_ZERO : Tests {
159    my $self = shift;
160    gsl_vector_set_zero($self->{vector});
161    map { is(gsl_vector_get($self->{vector} , $_ ), 0) } (0..4);
164 sub GSL_VECTOR_SET_BASIS : Tests {
165    my $self = shift;
166    ok_status(gsl_vector_set_basis($self->{vector}, 0));
167    is (gsl_vector_get($self->{vector} , 0 ), 1);
168    map { is(gsl_vector_get($self->{vector} , $_ ), 0) } (1..4);
171 sub GSL_VECTOR_SUBVECTOR_WITH_STRIDE : Tests {
172    my $self = shift;
173    map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
174    my $sub_stride = gsl_vector_subvector_with_stride($self->{vector}, 0, 2, 3);
175    is(gsl_vector_get($sub_stride->{vector} , 0 ), 0, "first element");
176    is(gsl_vector_get($sub_stride->{vector} , 1 ), 4, "second element");
177    is(gsl_vector_get($sub_stride->{vector} , 2 ), 16, "third element");
180 sub GSL_VECTOR_MAX_INDEX : Tests {
181    my $self = shift;
182    map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
183    my $index = gsl_vector_max_index($self->{vector});
184    is($index, 4, "Position of the maximum"); 
187 sub GSL_VECTOR_MIN_INDEX : Tests {
188    my $self = shift;
189    map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
190    my $index = gsl_vector_min_index($self->{vector});
191    is($index, 0, "Position of the minimum"); 
194 sub GSL_VECTOR_MINMAX_INDEX : Tests {
195    my $self = shift;
196    my ($min, $max);
197    map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); 
198    ($min, $max) = gsl_vector_minmax_index($self->{vector});
199    ok_similar( [ 0, 4 ], [ $min, $max], 'gsl_vector_minmax_index' );
202 sub GSL_VECTOR_MINMAX : Tests {
203    my ($min, $max);
204    my $vector = gsl_vector_alloc(5); 
205    map { gsl_vector_set($vector, $_, $_ ** 2 ) } (0..4); 
207    ($min, $max) = gsl_vector_minmax($vector);
209    ok_similar( [ 0, 16 ], [ $min, $max], 'gsl_vector_minmax' );
212 sub GSL_VECTOR_MEMCPY : Tests {
213    my $self = shift;
214    my $copy = gsl_vector_alloc(5);
215    map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
216    ok_status( gsl_vector_memcpy($copy, $self->{vector}) );
217    map { is(gsl_vector_get($copy, $_), $_ ** 2 ) } (0..4); ;
220 sub GSL_VECTOR_VIEW_ARRAY : Tests {
221    my @array = [1,2,3,4,5,6]; 
222    my $vec_view = gsl_vector_view_array(@array, 2);
223    map { is(gsl_vector_get($vec_view->{vector}, $_), $_+1 ) } (0..1); ;
226 sub GSL_VECTOR_REVERSE : Tests {
227    my $self = shift;
228    map { gsl_vector_set($self->{vector}, $_, $_ ** 2 ) } (0..4); ;
229    ok_status( gsl_vector_reverse($self->{vector}));
230    ok_similar( [ 16, 9, 4, 1, 0], [ map { gsl_vector_get($self->{vector}, $_) } 0..4 ] );
233 sub GSL_VECTOR_SWAP_ELEMENTS : Tests {
234    my $v1 = Math::GSL::Vector->new( [ map { $_ ** 2 } (0 .. 4) ] );
235    ok_status( gsl_vector_swap_elements($v1->raw, 0, 4));
237    is(gsl_vector_get($v1->raw, 0), 16);
238    is(gsl_vector_get($v1->raw, 4), 0);
240    is_deeply( [ 16, 1, 4, 9, 0 ], [ $v1->as_list ] );
243 sub GSL_VECTOR_ADD : Tests {
244    my $v1 = Math::GSL::Vector->new([0 .. 4]);
245    my $v2 = $v1->copy;
246    ok_status( gsl_vector_reverse($v2->raw) );
247    ok_status( gsl_vector_add($v1->raw, $v2->raw) );
248    is_deeply( [ $v1->as_list ], [ (4) x 5 ] );
251 sub GSL_VECTOR_SUB : Tests {
252    my $self = shift;
253    my $second_vec = gsl_vector_alloc(5);
254    map { gsl_vector_set($self->{vector}, $_, $_ ) } (0..4); ;
255    map { gsl_vector_set($second_vec, $_, 1) } (0..4); ;
256    ok_status( gsl_vector_sub($self->{vector}, $second_vec));
257    map { is(gsl_vector_get($self->{vector}, $_), $_ - 1 ) } (0..4); ;   
260 sub GSL_VECTOR_MUL : Tests {
261    my $self = shift;
262    my $second_vec = gsl_vector_alloc(5);
263    map { gsl_vector_set($self->{vector}, $_, $_ ) } (0..4); ;
264    map { gsl_vector_set($second_vec, $_, 2) } (0..4); ;
265    ok_status( gsl_vector_mul($self->{vector}, $second_vec));
266    map { is(gsl_vector_get($self->{vector}, $_), $_ * 2 ) } (0..4); ;   
269 sub GSL_VECTOR_DIV : Tests {
270    my $self = shift;
271    my $second_vec = gsl_vector_alloc(5);
272    map { gsl_vector_set($self->{vector}, $_, $_*2 ) } (0..4); ;
273    map { gsl_vector_set($second_vec, $_, 2) } (0..4); ;
274    ok_status( gsl_vector_div($self->{vector}, $second_vec));
275    map { is(gsl_vector_get($self->{vector}, $_), $_ ) } (0..4); ;   
278 sub GSL_VECTOR_SCALE : Tests {
279    my $v = Math::GSL::Vector->new([0..4]);
280    ok_status(gsl_vector_scale($v->raw, 2));
281    ok_similar( [ $v->as_list ], [ 0,2,4,6,8 ] );
284 sub GSL_VECTOR_SCALE_OVERLOAD : Tests {
285    my $v = Math::GSL::Vector->new([0..4]);
286    my $expected = [ map { $_*5} (0..4) ];
287    $v = 5 * $v;
288    ok_similar( [ $v->as_list ],  $expected );
290    my $w = Math::GSL::Vector->new([0..4]);
291    $w = $w * 5;
292    ok_similar( [ $w->as_list ], $expected );
295 sub GSL_VECTOR_DOT_PRODUCT : Tests {
296    my $v = Math::GSL::Vector->new([0..4]);
297    my $w = Math::GSL::Vector->new([0..4]);
299    ok_similar( $v * $w ,  4*4 + 3*3 + 2*2 + 1*1, 'basic dot product');
301    my $z = Math::GSL::Vector->new([0..10]);
302    dies_ok( sub { $z * $v; }, 'dot_product checks vector length' );
304    my $q = Math::GSL::Vector->new(5);
305    ok_similar ( $q * $q, 0, 'newly created vectors are zero-filled');
309 sub GSL_VECTOR_SWAP : Tests {
311    my @idx  = (0..(5+int rand(5)));
312    my $vec1 = gsl_vector_alloc($#idx+1);
313    my $vec2 = gsl_vector_alloc($#idx+1);
315    map { gsl_vector_set($vec1, $_, $_**2 ) } @idx;
316    map { gsl_vector_set($vec2, $_, $_)     } @idx;
318    ok_status( gsl_vector_swap($vec1, $vec2));
320    ok_similar( [ map { gsl_vector_get($vec1, $_)  } @idx ], 
321                [ @idx ],
322              );
323    ok_similar( [ map { gsl_vector_get($vec2, $_)  } @idx ], 
324                [ map { $_**2 } @idx ],
325              );
328 sub GSL_VECTOR_FPRINTF_FSCANF : Tests {  
329    my $vec1 = Math::GSL::Vector->new([ map { $_ ** 2 } (0..4) ]);
331    my $fh = gsl_fopen("vector", 'w');
332    ok( defined $fh, 'fopen -  write');
333    ok_status(gsl_vector_fprintf($fh, $vec1->raw, "%f"));
334    ok_status(gsl_fclose($fh));
336    my $vec2 = Math::GSL::Vector->new([ map { $_ ** 3 } (0..4) ]);
338    $fh = gsl_fopen("vector", 'r');   
339    ok( defined $fh, 'fopen  - readonly');
340    
341    ok_status(gsl_vector_fscanf($fh, $vec2->raw));
343    ok_similar( [ $vec2->as_list ], [ map { $_ ** 2 } (0..4) ]);
344    ok_status(gsl_fclose($fh) ); 
347 sub GSL_ADDITION : Tests {
348   my $vec1 = Math::GSL::Vector->new([1,2,3]);
349   my $vec2 = Math::GSL::Vector->new([2,3,4]);
350   my $vec3 = $vec1 + $vec2;
351   ok_similar([$vec3->as_list], [3,5,7]);
354   my $vec4 = $vec2 + 5;
355   ok_similar([$vec4->as_list], [7,8,9]);
357   my $vec5 = 5 + $vec2;
358   ok_similar([$vec5->as_list], [7,8,9]);
359    
360   my $z = Math::GSL::Vector->new([0..10]);
361   dies_ok( sub { $z + $vec1; }, 'addition checks vector length' );
362   ok_similar([$vec1->as_list], [1,2,3]);
365 sub COPY : Tests {
366     my $v1 = Math::GSL::Vector->new( [ 55 .. 65 ] );
367     isa_ok( $v1->copy, 'Math::GSL::Vector' );
368     ok_similar( [ $v1->copy->as_list ], [ $v1->as_list ] );
371 sub GSL_SUBTRACTION : Tests(3) {
372     my $v1 = Math::GSL::Vector->new( [ 1 .. 5 ]);
373     my $v2 = Math::GSL::Vector->new( [ 5 .. 9 ]);
375     ok_similar( [($v2-$v1)->as_list], [ (4) x 5 ] );
376     
377     my $v3 = $v2 - 3;
378     ok_similar( [ $v3->as_list  ], [ 2 .. 6 ] );
380     my $v4 = 4 - $v3;
381     ok_similar( [ $v4->as_list ], [ 2, 1, 0, -1, -2 ] );
384 sub GSL_MULTIPLICATION : Tests(6) {
385     my $v = Math::GSL::Vector->new([1,2,3]);
386     my $v2 = $v * 5;
388     # check that original is not modified each time
389     ok_similar ( [$v2->as_list], [5,10,15]);
390     ok_similar ( [$v->as_list],  [1,2,3]);
392     my $v3 = 5 * $v;
393     ok_similar ( [$v3->as_list], [5,10,15]);
394     ok_similar ( [$v->as_list],  [1,2,3]);
396     my $w = $v3 * 0;
397     ok_similar( [ $w->as_list ], [0,0,0], 'right overloaded zero-ify' );
399     my $z = 0 * $v3;
400     ok_similar( [ $w->as_list ], [0,0,0], 'left overloaded zero-ify' );
403 sub GSL_VECTOR_SWAP_OBJECTS : Tests(3) {
404     my $v = Math::GSL::Vector->new([1,2,3]);
405     my $w = Math::GSL::Vector->new([42,69,18]);
407     isa_ok( $v->swap($w), 'Math::GSL::Vector' );
409     ok_similar( [ $v->as_list ], [ 42, 69, 18 ] );
410     ok_similar( [ $w->as_list ], [ 1, 2, 3    ] );
413 sub GSL_VECTOR_REVERSE_OBJECTS : Tests(4) {
414     my @elements = map { int(rand(100)) } (1..10);
415     my $v        = Math::GSL::Vector->new([@elements]);
417     ok_similar( [ $v->reverse->as_list ], [ reverse @elements ] );
418     isa_ok($v->reverse, 'Math::GSL::Vector');
419     ok_similar( [ $v->as_list ], [ @elements ] );
420     ok_similar( [ $v->reverse->reverse->as_list ], [ $v->as_list ] );
423 sub GSL_VECTOR_NORM : Tests(5) {
424     my $v        = Math::GSL::Vector->new([ (0) x 5 ]);
425     ok_similar( $v->norm, 0 , 'zero vector norm = 0' );
427     isa_ok( $v->normalize, 'Math::GSL::Vector');
428     my $w = Math::GSL::Vector->new([1,2,3]);
429     my $z = Math::GSL::Vector->new([1,2,-3]);
430     ok_similar( [ $w->norm    ],    [ sqrt(14) ],  '2-norm' );
431     ok_similar( [ $w->norm(1) ],    [ 6        ],  '1-norm' );
433     ok_similar( [ $z->norm(1) ],    [ 6        ],  '1-norm with neg. elems.' );
437 sub GSL_VECTOR_NORMALIZE : Tests(4) {
438     my $w = Math::GSL::Vector->new([1,2,3]);
439     isa_ok( $w->normalize, 'Math::GSL::Vector');
440     ok_similar( $w->norm , 1, 'normalize default p=2');
442     my $v = Math::GSL::Vector->new([1,2,3]);
443     isa_ok( $v->normalize(3), 'Math::GSL::Vector' );
444     ok_similar ( $v->norm(3), 1, 'normalize p=3');
447 sub GSL_VECTOR_NORM_OVERLOAD: Tests(2) {
448     my $v = Math::GSL::Vector->new([1,2,3]);
449     my $w = Math::GSL::Vector->new([10,0,1]);
451     ok_similar( [ $v->norm ], [ abs $v ], 'abs overload');
452     ok_similar( [ abs $w   ], [ sqrt(101) ], 'abs = norm' )
455 Test::Class->runtests;