Store Sequence sequence location in a shared_ptr class
[mussa.git] / alg / test / test_sequence.cpp
index 48492f79294fd9535a10210f3522a62f0a608d2f..3847252a80884fa06786afe18feb7cf7f387a1e4 100644 (file)
@@ -1,3 +1,4 @@
+#define BOOST_AUTO_TEST_MAIN
 #include <boost/test/auto_unit_test.hpp>
 #include <boost/filesystem/path.hpp>
 #include <boost/filesystem/operations.hpp>
@@ -26,6 +27,24 @@ BOOST_AUTO_TEST_CASE( sequence_get_sequence )
        BOOST_CHECK_EQUAL(s.get_sequence(), std::string() );
 }
 
+BOOST_AUTO_TEST_CASE( sequence_from_string )
+{
+  std::string str1("AAAT");
+  Sequence seq1(str1);
+  BOOST_CHECK_EQUAL(seq1.get_sequence(), str1);
+} 
+
+BOOST_AUTO_TEST_CASE( sequence_find_first_not_of )
+{
+  std::string str1("AAAAT");
+  Sequence seq1(str1);
+  BOOST_CHECK_EQUAL(seq1.find_first_not_of("A"), str1.find_first_not_of("A"));
+  
+  std::string str2("AATTGGCC");
+  Sequence seq2(str2);
+  BOOST_CHECK_EQUAL(seq2.find_first_not_of("qwer"), str2.find_first_not_of("qwer"));
+}
+
 //! when we try to load a missing file, do we get an error?
 BOOST_AUTO_TEST_CASE( sequence_load_exception )
 {
@@ -178,6 +197,52 @@ BOOST_AUTO_TEST_CASE( sequence_load )
                                     "5' flank");
 }
 
+BOOST_AUTO_TEST_CASE( sequence_load_fasta_error )
+{
+  fs::path seq_path(fs::path(EXAMPLE_DIR, fs::native)/"seq");
+  seq_path /= "broken.fa";
+  bool exception_thrown = false;
+  std::string exception_filename;
+  Sequence s;
+  try {
+    s.load_fasta(seq_path);
+  } catch(sequence_invalid_load_error e) {
+    exception_thrown = true;
+    size_t native_string_size = seq_path.native_file_string().size();
+    std:string estr(e.what());
+    BOOST_REQUIRE(estr.size() > native_string_size);
+    std::copy(estr.begin(), estr.begin()+native_string_size,
+              std::back_inserter(exception_filename));
+  }
+  BOOST_CHECK_EQUAL(exception_thrown, true);
+  BOOST_CHECK_EQUAL(seq_path.native_file_string(), exception_filename);
+}
+
+BOOST_AUTO_TEST_CASE( sequence_load_annot_error )
+{
+  fs::path seq_path(fs::path(EXAMPLE_DIR, fs::native)/"seq");
+  seq_path /= "mouse_mck_pro.fa";
+  fs::path annot_path(fs::path(EXAMPLE_DIR, fs::native));
+  annot_path /= "broken.annot";
+  bool exception_thrown = false;
+  Sequence s;
+  s.load_fasta(seq_path);
+
+  std::string exception_filename;
+  try {
+    s.load_annot(annot_path, 0, 0);
+  } catch(annotation_load_error e) {
+    exception_thrown = true;
+    std:string estr(e.what());
+    size_t native_string_size = annot_path.native_file_string().size();
+    BOOST_REQUIRE(estr.size() > native_string_size);
+    std::copy(estr.begin(), estr.begin()+native_string_size,
+              std::back_inserter(exception_filename));
+  }
+  BOOST_CHECK_EQUAL(exception_thrown, true);
+  BOOST_CHECK_EQUAL(annot_path.native_file_string(), exception_filename);
+}
+
 BOOST_AUTO_TEST_CASE( sequence_load_dna_reduced )
 {
   std::string reduced_dna_fasta_string(">foo\nAAGGCCTTNN\n");
@@ -258,6 +323,32 @@ BOOST_AUTO_TEST_CASE( sequence_load_fasta_default )
                     sequence_invalid_load_error);
 }
 
+BOOST_AUTO_TEST_CASE( sequence_load_multiple_sequences_one_fasta ) 
+{
+  std::string fasta_file(
+    ">gi|10129974|gb|AF188002.1|AF188002\n"
+    "AAAAGGCTCCTGTCATATTGTGTCCTGCTCTGGTCTGC\n"
+    ">gi|180579|gb|M21487.1|HUMCKMM1\n"
+    "CGCCGAGAGCGCTTGCTCTGCCCAGATCTCGGCGAGTC\n"
+    ">gi|1621|emb|X55146.1|OCMCK1\n"
+    "CTCCCTGAGGGGAGTGCCCCGCTTAGCCC\n"
+  );
+  istringstream seq1_file(fasta_file);
+  Sequence seq1;
+  seq1.load_fasta(seq1_file, 1, 0, 0);
+  BOOST_CHECK_EQUAL(seq1.get_sequence(), "AAAAGGCTCCTGTCATATTGTGTCCTGCTCTGGTCTGC");
+  
+  istringstream seq2_file(fasta_file);
+  Sequence seq2;
+  seq2.load_fasta(seq2_file, 2, 0, 0);
+  BOOST_CHECK_EQUAL(seq2.get_sequence(), "CGCCGAGAGCGCTTGCTCTGCCCAGATCTCGGCGAGTC");
+  
+  istringstream seq3_file(fasta_file);
+  Sequence seq3;
+  seq3.load_fasta(seq3_file, 3, 0, 0);  
+  BOOST_CHECK_EQUAL(seq3.get_sequence(), "CTCCCTGAGGGGAGTGCCCCGCTTAGCCC"); 
+}
+
 BOOST_AUTO_TEST_CASE( sequence_reverse_complement )
 {
   std::string iupac_symbols("AaCcGgTtRrYySsWwKkMmBbDdHhVvNn-~.?");
@@ -287,6 +378,56 @@ BOOST_AUTO_TEST_CASE( sequence_reverse_complement_rna )
   BOOST_CHECK_EQUAL(rna_seq, rna_seq.rev_comp().rev_comp());
 }
 
+BOOST_AUTO_TEST_CASE( sequence_reverse_complement_subseq )
+{
+  std::string dna_str("AAAAAAAAAAGGGGGGGGGGG");
+  Sequence seq(dna_str, Sequence::reduced_dna_alphabet);
+  Sequence subseq = seq.subseq(8,4);
+  BOOST_CHECK_EQUAL( subseq, "AAGG");
+  Sequence rev_subseq = subseq.rev_comp();
+  BOOST_CHECK_EQUAL( rev_subseq.size(), subseq.size() );
+  BOOST_CHECK_EQUAL( rev_subseq.get_sequence(), "CCTT");
+}
+
+BOOST_AUTO_TEST_CASE( sequence_reverse_iterator )
+{
+  std::string dna_str("AAAAAAAAAAGGGGGGGGGGG");
+  std::string dna_str_reversed(dna_str.rbegin(), dna_str.rend());
+  Sequence seq(dna_str);
+  std::string seq_reversed(seq.rbegin(), seq.rend());
+  BOOST_CHECK_EQUAL(seq_reversed, dna_str_reversed);
+  
+  std::string substr = dna_str.substr(8,4);
+  Sequence subseq = seq.subseq(8,4);
+  BOOST_CHECK_EQUAL(substr, subseq);
+
+  std::string substr_reversed(substr.rbegin(), substr.rend());
+  std::string subseq_reversed(subseq.rbegin(), subseq.rend());
+  BOOST_CHECK_EQUAL(substr_reversed, subseq_reversed);  
+}
+
+BOOST_AUTO_TEST_CASE( sequence_empty_reverse_iterator)
+{
+  // so what happens with reverse interators when we have no sequence?
+  Sequence seq1;
+  Sequence seq2;
+  Sequence seq3("AGCT");
+  
+  // all the empty sequences should have equal iterators
+  BOOST_CHECK(seq1.rbegin() == seq1.rend());
+  BOOST_CHECK(seq1.rbegin() == seq2.rend());
+  
+  // none of the seq1 iterators should equal any of the seq3 iterators
+  BOOST_CHECK(seq1.rbegin() != seq3.rbegin());
+  BOOST_CHECK(seq1.rbegin() != seq3.rend());
+  BOOST_CHECK(seq1.rend() != seq3.rbegin());
+  BOOST_CHECK(seq1.rend() != seq3.rend());
+  
+  // seq3 iterators should work
+  BOOST_CHECK(seq3.rbegin() != seq3.rend());
+  
+}
+
 BOOST_AUTO_TEST_CASE( annotation_load )
 {
   string annot_data = "human\n"
@@ -536,6 +677,37 @@ BOOST_AUTO_TEST_CASE( sequence_motifs )
   */
 }
 
+BOOST_AUTO_TEST_CASE( sequence_motif_subseq)
+{
+  // when searching for a motif on a subsequence we should 
+  // only search the subsequence ticket:199
+  string aaaa("AAAA");
+  string cccc("CCCC");
+  Sequence s1("AAAANCCCC", Sequence::reduced_dna_alphabet);
+
+  // this shouldn't show up
+  s1.add_motif(cccc);
+  BOOST_CHECK_EQUAL( s1.motifs().size(), 1 );
+
+  s1.add_motif(aaaa);
+  BOOST_CHECK_EQUAL( s1.motifs().size(), 2 );
+
+  Sequence subseq1 = s1.subseq(4,5);
+  BOOST_CHECK_EQUAL(subseq1.motifs().size(), 2);
+  subseq1.clear_motifs();
+  BOOST_CHECK_EQUAL(subseq1.motifs().size(), 0);
+  // this is outside of our subsequence, and so shouldn't be found    
+  subseq1.add_motif(aaaa);
+  BOOST_CHECK_EQUAL( subseq1.motifs().size(), 0 );
+  
+  subseq1.add_motif(cccc);
+  BOOST_CHECK_EQUAL( subseq1.motifs().size(), 1);
+  std::list<motif>::const_iterator motif_i = subseq1.motifs().begin();
+  BOOST_REQUIRE(motif_i != subseq1.motifs().end());
+  BOOST_CHECK_EQUAL(motif_i->begin, 1);
+  BOOST_CHECK_EQUAL(motif_i->end, 5);
+}
+
 BOOST_AUTO_TEST_CASE( annot_test )
 {
   annot a(0, 10, "test", "thing");
@@ -793,5 +965,5 @@ BOOST_AUTO_TEST_CASE( serialize_xml_two )
   BOOST_CHECK_EQUAL(seq1_loaded, seq1);
   BOOST_CHECK_EQUAL(seq2_loaded, seq2);
   // test if our pointers are the same
-  BOOST_CHECK_EQUAL(seq1_loaded.c_str(), seq2_loaded.c_str());
+  BOOST_CHECK_EQUAL(seq1_loaded.data(), seq2_loaded.data());
 }