remove annot class in favor of SeqSpan

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

diff --git a/py/test/TestSequence.py b/py/test/TestSequence.py
index 23cdc93e22b3528616ae9ce2498e90845826d344..80fc3e68e84f5489169b011207d489c0023049a5 100644 (file)
--- a/py/test/TestSequence.py
+++ b/py/test/TestSequence.py
@@ -14,7 +14,34 @@ class TestSequence(unittest.TestCase):
     self.failUnless(len(seq_text) == len(s))
     self.failUnless(len(s) == s.size())
     self.failUnless(str(s) == seq_text)
- 
+    self.failUnless(str(s) == seq_text)
+
+  def testNames(self):
+    seq_text = "AAGGCCTT"
+    header_text = "hello"
+    species = "Goo Gooerific"
+
+    s = mussa.Sequence(seq_text)
+
+    s.header = header_text
+    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')