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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
|
#include<iostream>
#include "../src/call_consensus_pileup.h"
#include "../src/allele_functions.h"
int main() {
int num_tests = 6;
allele a1 = {
"A",
6,
0,
30,
0,
0,
0
};
allele ai1 = {
"+T",
3,
0,
30,
0,
0,
0
};
allele a2 = {
"T",
1,
10,
30,
0,
0,
0
};
allele a3 = {
"C",
1,
10,
30,
0,
0,
0
};
allele a4 = {
"G",
1,
10,
30,
0,
0,
0
};
allele a5 = {
"*",
1,
10,
30,
0,
0,
0
};
allele a6 = {
"G",
2,
10,
30,
0,
0,
0
};
int success = 0;
allele arr[] = {a1, ai1,a2,a3,a4,a5};
ret_t s;
std::vector<allele> ad(arr, arr+sizeof(arr)/sizeof(allele));
int size = sizeof(arr)/sizeof(allele);
for(int i = 0;i<size;i++){
ad.at(i) = arr[i];
}
s = get_consensus_allele(ad,20,.6, 'N');
std::cout << s.nuc << ": " << s.q << std::endl;
success += (s.nuc.compare("A") == 0) ? 1: 0;
success += (s.q.compare("?") == 0) ? 1 : 0;
s = get_consensus_allele(ad,20,.7, 'N');
std::cout << s.nuc << ": " << s.q << std::endl;
success += (s.nuc.compare("N") == 0) ? 1: 0;
success += (s.q.compare("?") == 0) ? 1 : 0;
ad.erase(ad.begin() + 4, ad.begin()+5);
ad.push_back(a6);
s = get_consensus_allele(ad,20,.7, 'N');
std::cout << s.nuc << ": " << s.q << std::endl;
success += (s.nuc.compare("R") == 0) ? 1: 0;
success += (s.q.compare("?") == 0) ? 1 : 0;
return (success == num_tests) ? 0 : -1;
}
|