def test_len(self):
"""Stem len() should return self.Length"""
s = Stem()
self.assertEqual(len(s), 0)
s.Length = 5
self.assertEqual(len(s), 5)
s.Length = None
self.assertRaises(TypeError, len, s)
def test_len(self):
"""Stem len() should return self.Length"""
s = Stem()
self.assertEqual(len(s), 0)
s.Length = 5
self.assertEqual(len(s), 5)
s.Length = None
self.assertRaises(TypeError, len, s)
def test_getitem(self):
"""Stem getitem should return a Stem object for the ith pair in the stem"""
s = Stem()
self.assertRaises(IndexError, s.__getitem__, 0)
s.Start = 5
s.End = 8
s.Length = 1
pairs = list(s)
self.assertEqual(pairs, [Stem(5, 8, 1)])
s.Length = 2
pairs = list(s)
self.assertEqual(pairs, [Stem(5, 8, 1), Stem(6, 7, 1)])
#WARNING: Stem will not complain when iterating over an invalid helix,
#as per the one below
s.Length = 5
pairs = list(s)
self.assertEqual(pairs, [Stem(5,8,1),Stem(6,7,1),Stem(7,6,1),\
Stem(8,5,1), Stem(9,4,1)])
def test_getitem(self):
"""Stem getitem should return a Stem object for the ith pair in the stem"""
s = Stem()
self.assertRaises(IndexError, s.__getitem__, 0)
s.Start = 5
s.End = 8
s.Length = 1
pairs = list(s)
self.assertEqual(pairs, [Stem(5, 8, 1)])
s.Length = 2
pairs = list(s)
self.assertEqual(pairs, [Stem(5,8,1),Stem(6,7,1)])
#WARNING: Stem will not complain when iterating over an invalid helix,
#as per the one below
s.Length = 5
pairs = list(s)
self.assertEqual(pairs, [Stem(5,8,1),Stem(6,7,1),Stem(7,6,1),\
Stem(8,5,1), Stem(9,4,1)])
