def test_model_gets_first_model(self):
pdb = Pdb()
pdb._models = ["1", "2", "3"]
self.assertEqual(pdb.model, "1")
def test_can_get_pdb_models(self):
pdb = Pdb()
pdb._models = ["1", "2", "3"]
self.assertEqual(pdb.models, ("1", "2", "3"))
def test_pdb_repr_multiple_models(self):
pdb = Pdb()
pdb._models = ["1", "2", "3"]
self.assertEqual(str(pdb), "<Pdb (3 models)>")
def test_pdb_repr_with_code(self):
pdb = Pdb()
pdb._code = "1XXX"
pdb._models = ["1", "2", "3"]
self.assertEqual(str(pdb), "<Pdb 1XXX (3 models)>")
def test_pdb_repr_one_model(self):
pdb = Pdb()
pdb._models = ["1"]
self.assertEqual(str(pdb), "<Pdb (1 model)>")