trim test cases now check sequence contents

CountESS-Project · Dec 18, 2019 · 0fa4144 · 0fa4144
1 parent e85f5f4
commit 0fa4144
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/test_fastqread.py b/tests/test_fastqread.py
@@ -111,11 +111,13 @@ def test_trim(self):
         test_read.trim(start=2)
         self.assertEqual(len(test_read), len(FastqRead(**self.test_kwargs)) - 1)
         self.assertEqual(len(test_read.sequence), len(test_read.quality))
+        self.assertEqual(test_read.sequence, FastqRead(**self.test_kwargs).sequence[1:])
 
         test_read = FastqRead(**self.test_kwargs)
         test_read.trim(start=len(test_read))
         self.assertEqual(len(test_read), 1)
         self.assertEqual(len(test_read.sequence), len(test_read.quality))
+        self.assertEqual(test_read.sequence, FastqRead(**self.test_kwargs).sequence[-1])
 
         # trim the end
         test_read = FastqRead(**self.test_kwargs)
@@ -126,15 +128,20 @@ def test_trim(self):
         test_read.trim(end=len(test_read) - 1)
         self.assertEqual(len(test_read), len(FastqRead(**self.test_kwargs)) - 1)
         self.assertEqual(len(test_read.sequence), len(test_read.quality))
+        self.assertEqual(test_read.sequence, FastqRead(**self.test_kwargs).sequence[:-1])
 
         test_read = FastqRead(**self.test_kwargs)
         test_read.trim(end=1)
         self.assertEqual(len(test_read), 1)
         self.assertEqual(len(test_read.sequence), len(test_read.quality))
+        self.assertEqual(test_read.sequence, FastqRead(**self.test_kwargs).sequence[:1])
 
         # trim both ends
-        # TODO
         test_read = FastqRead(**self.test_kwargs)
+        test_read.trim(start=2, end=2)
+        self.assertEqual(len(test_read), 1)
+        self.assertEqual(len(test_read.sequence), len(test_read.quality))
+        # TODO
 
         # bad start parameters
         self.assertRaises(ValueError, test_read.trim, start=-1)