source: tests/src/test-cvec.c @ 9e2c1a1

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/timestretchfix/ffmpeg5
Last change on this file since 9e2c1a1 was dfd520b, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[tests] improve test-cvec

  • Property mode set to 100644
File size: 2.1 KB
Line 
1#include "aubio.h"
2#include "utils_tests.h"
3
4int main (void)
5{
6  uint_t i, window_size = 16;
7  cvec_t * complex_vector = new_cvec(window_size);
8  cvec_t * other_cvector = new_cvec(window_size);
9
10  assert(cvec_norm_get_data(complex_vector) == complex_vector->norm);
11  assert(cvec_phas_get_data(complex_vector) == complex_vector->phas);
12  assert(complex_vector->length == window_size / 2 + 1);
13
14  // all elements are initialized to 0
15  for ( i = 0; i < complex_vector->length; i++ ) {
16    assert( complex_vector->norm[i] == 0. );
17    assert( complex_vector->phas[i] == 0. );
18  }
19
20  cvec_norm_set_sample(complex_vector, 2., 1);
21  assert(cvec_norm_get_sample(complex_vector, 1));
22
23  cvec_phas_set_sample(complex_vector, 2., 1);
24  assert(cvec_phas_get_sample(complex_vector, 1));
25
26  cvec_print(complex_vector);
27
28  // set all norm and phas elements to 0
29  cvec_zeros(complex_vector);
30  for ( i = 0; i < complex_vector->length; i++ ) {
31    assert( complex_vector->norm[i] == 0. );
32    assert( complex_vector->phas[i] == 0. );
33  }
34
35  // set all norm elements to 1
36  cvec_norm_ones(complex_vector);
37  for ( i = 0; i < complex_vector->length; i++ ) {
38    assert( complex_vector->norm[i] == 1. );
39  }
40
41  // set all norm elements to 0
42  cvec_norm_zeros(complex_vector);
43  for ( i = 0; i < complex_vector->length; i++ ) {
44    assert( complex_vector->norm[i] == 0. );
45  }
46
47  // set all phas elements to 1
48  cvec_phas_ones(complex_vector);
49  for ( i = 0; i < complex_vector->length; i++ ) {
50    assert( complex_vector->phas[i] == 1. );
51  }
52
53  // set all phas elements to 0
54  cvec_phas_zeros(complex_vector);
55  for ( i = 0; i < complex_vector->length; i++ ) {
56    assert( complex_vector->phas[i] == 0. );
57  }
58
59  cvec_copy(complex_vector, other_cvector);
60  // copy to self
61  cvec_copy(complex_vector, complex_vector);
62  // copy to a different size fails
63  del_cvec(other_cvector);
64  other_cvector = new_cvec(window_size + 2);
65  cvec_copy(complex_vector, other_cvector);
66
67  if (complex_vector)
68    del_cvec(complex_vector);
69  if (other_cvector)
70    del_cvec(other_cvector);
71
72  // wrong parameters
73  assert(new_cvec(-1) == NULL);
74  assert(new_cvec(0) == NULL);
75
76  return 0;
77}
Note: See TracBrowser for help on using the repository browser.