Skip to content
Snippets Groups Projects
Commit f950e5a7 authored by Bienchen's avatar Bienchen
Browse files

Exported InChI from compoundslib to Python

parent 5b0e6de4
No related branches found
No related tags found
No related merge requests found
...@@ -106,7 +106,7 @@ void export_Compound() { ...@@ -106,7 +106,7 @@ void export_Compound() {
.def("GetAtomSpecs", &Compound::GetAtomSpecs, .def("GetAtomSpecs", &Compound::GetAtomSpecs,
return_value_policy<copy_const_reference>()) return_value_policy<copy_const_reference>())
.add_property("bond_specs", make_function(&Compound::GetBondSpecs, .add_property("bond_specs", make_function(&Compound::GetBondSpecs,
return_value_policy<copy_const_reference>())) return_value_policy<copy_const_reference>()))
.add_property("atom_specs", make_function(&Compound::GetAtomSpecs, .add_property("atom_specs", make_function(&Compound::GetAtomSpecs,
return_value_policy<copy_const_reference>())) return_value_policy<copy_const_reference>()))
.def("AddAtom", &Compound::AddAtom) .def("AddAtom", &Compound::AddAtom)
...@@ -120,6 +120,12 @@ void export_Compound() { ...@@ -120,6 +120,12 @@ void export_Compound() {
&Compound::SetFormula) &Compound::SetFormula)
.add_property("dialect", &Compound::GetDialectAsString, .add_property("dialect", &Compound::GetDialectAsString,
&set_dialect) &set_dialect)
.add_property("inchi",
make_function(&Compound::GetInchi,
return_value_policy<copy_const_reference>()))
.add_property("inchi_key",
make_function(&Compound::GetInchiKey,
return_value_policy<copy_const_reference>()))
; ;
class_<AtomSpec>("AtomSpec", no_init) class_<AtomSpec>("AtomSpec", no_init)
......
...@@ -29,6 +29,9 @@ class TestCompound(unittest.TestCase): ...@@ -29,6 +29,9 @@ class TestCompound(unittest.TestCase):
self.assertEqual(compound.dialect, 'PDB') self.assertEqual(compound.dialect, 'PDB')
self.assertEqual(compound.formula, 'C3 H7 N O2') self.assertEqual(compound.formula, 'C3 H7 N O2')
self.assertEqual(compound.chem_class, mol.L_PEPTIDE_LINKING) self.assertEqual(compound.chem_class, mol.L_PEPTIDE_LINKING)
self.assertEqual(compound.inchi,
"1S/C3H7NO2/c1-2(4)3(5)6/h2H,4H2,1H3,(H,5,6)/t2-/m0/s1")
self.assertEqual(compound.inchi_key, "QNAYBMKLOCPYGJ-REOHCLBHSA-N")
if __name__=='__main__': if __name__=='__main__':
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment