def testAppend(self): chain = Chain(self.a[:]) chain.append('U') elements = self.a.strip().split() num_elements = len(elements) last_element = chain.data[-1] self.assertEqual('u', last_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.a[:]) chain.append(Hetero('A')) elements = self.a.strip().split() num_elements = len(elements) last_element = chain.data[-1] self.assertEqual('a', last_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.b[:]) chain.append('t') elements = self.b.strip().split() num_elements = len(elements) last_element = chain.data[-1] self.assertEqual('t', last_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.b[:]) chain.append(Hetero('C')) elements = self.b.strip().split() num_elements = len(elements) last_element = chain.data[-1] self.assertEqual('c', last_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.c[:]) chain.append('ser') elements = self.c.strip().split() num_elements = len(elements) last_element = chain.data[-1] self.assertEqual('ser', last_element.data) self.assertEqual(len(chain), num_elements + 1)
def testInsert(self): chain = Chain(self.a[:]) i = 4 chain.insert(i, "g") elements = self.a.strip().split() num_elements = len(elements) target_element = chain.data[i] self.assertEqual("g", target_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.a[:]) i = 0 chain.insert(i, "t") elements = self.a.strip().split() num_elements = len(elements) target_element = chain.data[i] self.assertEqual("t", target_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.b[:]) i = 9 chain.insert(i, Hetero("a")) elements = self.a.strip().split() num_elements = len(elements) target_element = chain.data[i] self.assertEqual("a", target_element.data) self.assertEqual(len(chain), num_elements + 1) chain = Chain(self.c[:]) i = 5 chain.insert(i, "gln") elements = self.c.strip().split() num_elements = len(elements) target_element = chain.data[i] self.assertEqual("gln", target_element.data) self.assertEqual(len(chain), num_elements + 1)
def testContains(self): chain = Chain(self.c[:]) self.assertFalse("ser" in chain) self.assertIn("lys", chain) self.assertIn("asp", chain)