Extend the python mussa interface.

[mussa.git] / py / test / TestSequence.py

diff --git a/py/test/TestSequence.py b/py/test/TestSequence.py
index 274fbdba4cd751da41dec9221b9cc79a888eff3a..e9d816e2665f7e844d9e84a1e57d16035530107a 100644 (file)
--- a/py/test/TestSequence.py
+++ b/py/test/TestSequence.py
@@ -16,6 +16,9 @@ class TestSequence(unittest.TestCase):
     self.failUnless(str(s) == seq_text)
     self.failUnless(str(s) == seq_text)
 
+    for i in range(len(seq_text)):
+      self.failUnlessEqual(s[i], seq_text[i])
+
   def testNames(self):
     seq_text = "AAGGCCTT"
     header_text = "hello"
@@ -27,7 +30,21 @@ class TestSequence(unittest.TestCase):
     s.species = species
     self.failUnless(s.header == header_text)
     self.failUnless(s.species == species)
-     
+
+  def testAnnotations(self):
+    seq = mussa.Sequence("AAGGCCTTAATTGGCCTT")
+    seq.add_annotation("foo", "utr", 0, 10)
+
+  def notestFile(self):
+    # remove no prefix once we have sequence loading from a stream
+    seq_text = "AAGGCCTT"
+    from StringIO import StringIO
+    s = StringIO(seq_text)
+
+    seq = mussa.Sequence(s)
+
+    self.failUnless(seq == seq_text)
+
 def suite():
   return unittest.makeSuite(TestSequence, 'test')