def test_add_del_pdb(self):
"""Add and delete a PDB entry."""
template_db = TemplateDatabase(str(self.database), str(self.file_root))
template_db.add_pdb("1del", self._METADATA)
template_db.commit()
self.assertEqual(template_db.get_pdb("1del"), self._METADATA)
template_db.del_pdb("1del")
template_db.commit()
with self.assertRaises(template_db.PdbNotFoundException):
template_db.get_pdb("1del")
def test_add_update_pdb(self):
"""Add and delete a PDB entry."""
template_db = TemplateDatabase(str(self.database), str(self.file_root))
template_db.add_pdb("1del", self._METADATA)
template_db.commit()
self.assertEqual(template_db.get_pdb("1del"), self._METADATA)
altered_meta = self._METADATA.copy()
altered_meta["resolution"] = 100
template_db.update_pdb("1del", altered_meta)
template_db.commit()
self.assertEqual(template_db.get_pdb("1del"), altered_meta)
def test_retrieve_pdb(self):
"""Retrieve PDB metadata from database."""
template_db = TemplateDatabase(str(self.database), str(self.file_root))
self.assertEqual(template_db.get_pdb("12as"), self._METADATA)