fixed current tests

WGS-TB · Jan 22, 2019 · b257b6f · b257b6f
1 parent bf4df71
commit b257b6f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1,922 deletions.
diff --git a/bin/prince b/bin/prince
@@ -11,7 +11,7 @@ from prince.boost import run_boosts
 from prince.query_sample import test_target
 
 DEFAULT_K = 25
-DEFAULT_BOOST_OUTPUT = resource_filename('prince.resources', 'training_data.txt')
+DEFAULT_BOOST_OUTPUT = resource_filename('prince.resources', 'training_data_w_extensions.txt')
 
 def main():
     parser = argparse.ArgumentParser(description='Prince Options.')

diff --git a/prince/tests/test_kmer_generator.py b/prince/tests/test_kmer_generator.py
@@ -6,4 +6,6 @@ class KmerGeneratorTest(unittest.TestCase):
 
     def test_kmer_generator(self):
         self.assertEqual(kmer_generator("", 0), {})
-        self.assertEqual(kmer_generator("A",1), {0: ['A']})
+        self.assertEqual(kmer_generator("A",1), {0: ['A', 'T']})
+        self.assertEqual(kmer_generator(["ATTCG","GGGG"],3), {0: ['ATT', 'TTC', 'TCG', 'CGA', 'GAT', 'ATC', 'TCG', 'CGA', 'GAA', 'AAT'], 1: ['GGG', 'GGG', 'GGG', 'GGG', 'CCC', 'CCC', 'CCC', 'CCC']})
+        self.assertEqual(kmer_generator(["ATTCG","GGGG"],3, extension=False), {0: ['ATT', 'TTC', 'TCG', 'CGA', 'GAA', 'AAT'], 1: ['GGG', 'GGG', 'CCC', 'CCC']})