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)])