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)
