class ResiduesTest(ut.TestCase):
def setUp(self):
self.structure = Structure([nts.nt77_9, nts.nt78_9, nts.nt79_9,
nts.nt80_9], pdb="1S72")
def test_can_get_sequence_from_all_residues(self):
self.assertEquals(['A', 'G', 'U', 'A'], self.structure.sequence)
def test_can_iterate_over_all_residues(self):
val = list(self.structure.residues())
self.assertEquals([nts.nt77_9, nts.nt78_9, nts.nt79_9, nts.nt80_9],
val)
def test_can_select_residues(self):
val = list(self.structure.residues(sequence='A'))
self.assertEquals([nts.nt77_9, nts.nt80_9], val)
def test_skip_iterating_over_nonpolymeric_by_default(self):
self.structure._residues.append(nts.nt212_0)
val = list(self.structure.residues())
self.assertEquals([nts.nt77_9, nts.nt78_9, nts.nt79_9, nts.nt80_9],
val)
def test_giving_polymeric_is_none_iterates_overall(self):
self.structure._residues.append(nts.nt212_0)
val = list(self.structure.residues(polymeric=None))
ans = [nts.nt77_9, nts.nt78_9, nts.nt79_9, nts.nt80_9, nts.nt212_0]
self.assertEquals(ans, val)
