def test_get_chains_protein(self): """Tests for get_chains function using protein pdb lines. """ chains_prot = get_chains(TEST_PDB_STRING_2.split('\n')) self.assertEqual(chains_prot, self.test_pdb_chains_2)
def test_get_chains_nucleotide(self): """Tests for get_chains function using nucleotide pdb lines. """ chains_nuc = get_chains(TEST_PDB_STRING_1.split('\n')) self.assertEqual(chains_nuc, self.test_pdb_chains_1)