File: view_trim_test.cpp

package info (click to toggle)
seqan3 3.0.2%2Bds-9
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 16,052 kB
  • sloc: cpp: 144,641; makefile: 1,288; ansic: 294; sh: 228; xml: 217; javascript: 50; python: 27; php: 25
file content (98 lines) | stat: -rw-r--r-- 4,638 bytes parent folder | download
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
94
95
96
97
98
// -----------------------------------------------------------------------------------------------------
// Copyright (c) 2006-2020, Knut Reinert & Freie Universität Berlin
// Copyright (c) 2016-2020, Knut Reinert & MPI für molekulare Genetik
// This file may be used, modified and/or redistributed under the terms of the 3-clause BSD-License
// shipped with this file and also available at: https://github.com/seqan/seqan3/blob/master/LICENSE.md
// -----------------------------------------------------------------------------------------------------

#include <iostream>

#include <gtest/gtest.h>

#include <seqan3/alphabet/quality/all.hpp>
#include <seqan3/range/concept.hpp>
#include <seqan3/range/views/to_char.hpp>
#include <seqan3/range/views/to.hpp>
#include <seqan3/range/views/trim_quality.hpp>
#include <seqan3/std/ranges>

using seqan3::operator""_dna5;

TEST(view_trim, standalone)
{
    std::vector<seqan3::phred42> vec{ seqan3::phred42{40}, seqan3::phred42{40},
                                      seqan3::phred42{30}, seqan3::phred42{20}, seqan3::phred42{10}};
    std::vector<seqan3::phred42> cmp1{seqan3::phred42{40}, seqan3::phred42{40},
                                      seqan3::phred42{30}, seqan3::phred42{20}};
    std::vector<seqan3::phred42> cmp2{seqan3::phred42{40}, seqan3::phred42{40}};

    // trim by phred_value
    auto v1 = vec | seqan3::views::trim_quality(20u);                   // == ['I','I','?','5']
    EXPECT_EQ(v1 | seqan3::views::to<std::vector>, cmp1);

    // trim by quality character
    auto v2 = vec | seqan3::views::trim_quality(seqan3::phred42{40});   // == ['I','I']
    EXPECT_EQ(v2 | seqan3::views::to<std::vector>, cmp2);

    // function syntax
    auto v3 = seqan3::views::trim_quality(vec, 20u);                    // == ['I','I','?','5']
    EXPECT_EQ(v3 | seqan3::views::to<std::vector>, cmp1);

    // combinability
    std::string v4 = seqan3::views::trim_quality(vec, 20u) | seqan3::views::to_char | seqan3::views::to<std::string>; //=="II?5"
    EXPECT_EQ("II?5", v4);
}

TEST(view_trim, qualified)
{
    std::vector<seqan3::dna5q> vec{{'A'_dna5, seqan3::phred42{40}},
                                   {'G'_dna5, seqan3::phred42{40}},
                                   {'G'_dna5, seqan3::phred42{30}},
                                   {'A'_dna5, seqan3::phred42{20}},
                                   {'T'_dna5, seqan3::phred42{10}}};
    std::vector<seqan3::dna5q> cmp1{{'A'_dna5, seqan3::phred42{40}},
                                    {'G'_dna5, seqan3::phred42{40}},
                                    {'G'_dna5, seqan3::phred42{30}},
                                    {'A'_dna5, seqan3::phred42{20}}};
    std::vector<seqan3::dna5q> cmp2{{'A'_dna5, seqan3::phred42{40}},
                                    {'G'_dna5, seqan3::phred42{40}}};

    // trim by phred_value
    auto v1 = vec | seqan3::views::trim_quality(20u);
    EXPECT_EQ(v1 | seqan3::views::to<std::vector>, cmp1);

    // trim by quality character
    auto v2 = vec | seqan3::views::trim_quality(seqan3::dna5q{'C'_dna5, seqan3::phred42{40}});
    EXPECT_EQ(v2 | seqan3::views::to<std::vector>, cmp2);

    // function syntax
    auto v3 = seqan3::views::trim_quality(vec, 20u);
    EXPECT_EQ(v3 | seqan3::views::to<std::vector>, cmp1);

    // combinability
    std::string v4 = seqan3::views::trim_quality(vec, 20u) | seqan3::views::to_char | seqan3::views::to<std::string>;
    EXPECT_EQ("AGGA", v4);
}

TEST(view_trim, concepts)
{
    std::vector<seqan3::dna5q> vec{{'A'_dna5, seqan3::phred42{40}},
                                   {'G'_dna5, seqan3::phred42{40}},
                                   {'G'_dna5, seqan3::phred42{30}},
                                   {'A'_dna5, seqan3::phred42{20}},
                                   {'T'_dna5, seqan3::phred42{10}}};
    EXPECT_TRUE(std::ranges::input_range<decltype(vec)>);
    EXPECT_TRUE(std::ranges::forward_range<decltype(vec)>);
    EXPECT_TRUE(std::ranges::random_access_range<decltype(vec)>);
    EXPECT_TRUE(std::ranges::common_range<decltype(vec)>);
    EXPECT_TRUE((std::ranges::output_range<decltype(vec), seqan3::dna5q>));
    EXPECT_TRUE(std::ranges::sized_range<decltype(vec)>);

    auto v1 = vec | seqan3::views::trim_quality(20u);
    EXPECT_TRUE(std::ranges::input_range<decltype(v1)>);
    EXPECT_TRUE(std::ranges::forward_range<decltype(v1)>);
    EXPECT_TRUE(std::ranges::random_access_range<decltype(v1)>);
    EXPECT_FALSE(std::ranges::common_range<decltype(v1)>);
    EXPECT_TRUE((std::ranges::output_range<decltype(v1), seqan3::dna5q>));
    EXPECT_TRUE(!std::ranges::sized_range<decltype(v1)>);
}