def testFromAdjacencyList3(self):
"""
adjlist: Test the Molecule.fromAdjacencyList() method 3.
"""
# molecule 3
adjlist = """
1 *1 C u1 {2,S} {3,S} {4,S}
2 H u0 {1,S}
3 H u0 {1,S}
4 *2 N u0 p0 c+1 {1,S} {5,S} {6,D}
5 O u0 p3 c-1 {4,S}
6 O u0 p2 {4,D}
"""
molecule = Molecule().fromAdjacencyList(adjlist)
self.assertTrue(molecule.multiplicity == 2)
atom1 = molecule.atoms[0]
atom2 = molecule.atoms[3]
atom3 = molecule.atoms[4]
atom4 = molecule.atoms[5]
self.assertTrue(molecule.hasBond(atom2, atom1))
self.assertTrue(molecule.hasBond(atom2, atom3))
self.assertTrue(molecule.hasBond(atom2, atom4))
self.assertFalse(molecule.hasBond(atom1, atom3))
self.assertFalse(molecule.hasBond(atom1, atom4))
bond21 = atom2.bonds[atom1]
bond23 = atom2.bonds[atom3]
bond24 = atom2.bonds[atom4]
self.assertTrue(atom1.label == '*1')
self.assertTrue(atom1.element.symbol == 'C')
self.assertTrue(atom1.radicalElectrons == 1)
self.assertTrue(atom1.charge == 0)
self.assertTrue(atom2.label == '*2')
self.assertTrue(atom2.element.symbol == 'N')
self.assertTrue(atom2.radicalElectrons == 0)
self.assertTrue(atom2.charge == 1)
self.assertTrue(atom3.label == '')
self.assertTrue(atom3.element.symbol == 'O')
self.assertTrue(atom3.radicalElectrons == 0)
self.assertTrue(atom3.charge == -1)
self.assertTrue(atom4.label == '')
self.assertTrue(atom4.element.symbol == 'O')
self.assertTrue(atom4.radicalElectrons == 0)
self.assertTrue(atom4.charge == 0)
self.assertTrue(bond21.isSingle())
self.assertTrue(bond23.isSingle())
self.assertTrue(bond24.isDouble())
def testFromAdjacencyList3(self):
"""
adjlist: Test the Molecule.fromAdjacencyList() method 3.
"""
# molecule 3
adjlist = """
1 *1 C u1 {2,S} {3,S} {4,S}
2 H u0 {1,S}
3 H u0 {1,S}
4 *2 N u0 p0 c+1 {1,S} {5,S} {6,D}
5 O u0 p3 c-1 {4,S}
6 O u0 p2 {4,D}
"""
molecule = Molecule().fromAdjacencyList(adjlist)
self.assertTrue(molecule.multiplicity == 2)
atom1 = molecule.atoms[0]
atom2 = molecule.atoms[3]
atom3 = molecule.atoms[4]
atom4 = molecule.atoms[5]
self.assertTrue(molecule.hasBond(atom2, atom1))
self.assertTrue(molecule.hasBond(atom2, atom3))
self.assertTrue(molecule.hasBond(atom2, atom4))
self.assertFalse(molecule.hasBond(atom1, atom3))
self.assertFalse(molecule.hasBond(atom1, atom4))
bond21 = atom2.bonds[atom1]
bond23 = atom2.bonds[atom3]
bond24 = atom2.bonds[atom4]
self.assertTrue(atom1.label == '*1')
self.assertTrue(atom1.element.symbol == 'C')
self.assertTrue(atom1.radicalElectrons == 1)
self.assertTrue(atom1.charge == 0)
self.assertTrue(atom2.label == '*2')
self.assertTrue(atom2.element.symbol == 'N')
self.assertTrue(atom2.radicalElectrons == 0)
self.assertTrue(atom2.charge == 1)
self.assertTrue(atom3.label == '')
self.assertTrue(atom3.element.symbol == 'O')
self.assertTrue(atom3.radicalElectrons == 0)
self.assertTrue(atom3.charge == -1)
self.assertTrue(atom4.label == '')
self.assertTrue(atom4.element.symbol == 'O')
self.assertTrue(atom4.radicalElectrons == 0)
self.assertTrue(atom4.charge == 0)
self.assertTrue(bond21.isSingle())
self.assertTrue(bond23.isSingle())
self.assertTrue(bond24.isDouble())
def test_atom_mapping_2(self):
"""Test that toRDKitMol returns correct indices and atom mappings when hydrogens are removed."""
adjlist = """
1 H u0 p0 c0 {2,S}
2 C u0 p0 c0 {1,S} {3,S} {4,S} {5,S}
3 H u0 p0 c0 {2,S}
4 H u0 p0 c0 {2,S}
5 O u0 p2 c0 {2,S} {6,S}
6 H u0 p0 c0 {5,S}
"""
mol = Molecule().fromAdjacencyList(adjlist)
rdkitmol, rdAtomIndices = toRDKitMol(mol,
removeHs=True,
returnMapping=True)
heavy_atoms = [at for at in mol.atoms if at.number != 1]
for at1 in heavy_atoms:
for at2 in heavy_atoms:
if mol.hasBond(at1, at2):
try:
rdkitmol.GetBondBetweenAtoms(rdAtomIndices[at1],
rdAtomIndices[at2])
except RuntimeError:
self.fail(
"RDKit failed in finding the bond in the original atom!"
)
