def test_isotope_subgraph_isomorphism_molecule_and_group():
"""
Checks that subgraph isomorphism works with enriched molecules and groups
"""
methanei = Molecule().fromAdjacencyList("""
1 C u0 p0 c0 i13 {2,S} {3,S} {4,S} {5,S}
2 H u0 p0 c0 {1,S}
3 H u0 p0 c0 {1,S}
4 H u0 p0 c0 {1,S}
5 H u0 p0 c0 {1,S}
""")
methane = Molecule().fromAdjacencyList("""
1 C u0 p0 c0 {2,S} {3,S} {4,S} {5,S}
2 H u0 p0 c0 {1,S}
3 H u0 p0 c0 {1,S}
4 H u0 p0 c0 {1,S}
5 H u0 p0 c0 {1,S}
""")
group_methane = Group().fromAdjacencyList("""
1 C u0 {2,S} {3,S} {4,S} {5,S}
2 H u0 {1,S}
3 H u0 {1,S}
4 H u0 {1,S}
5 H u0 {1,S}
""")
assert_true(methanei.isSubgraphIsomorphic(group_methane))
assert_true(methane.isSubgraphIsomorphic(group_methane))
assert_false(methanei.isIsomorphic(methane))
def test_isotope_subgraph_isomorphism_molecule_and_group():
"""
Checks that subgraph isomorphism works with enriched molecules and groups
"""
methanei = Molecule().fromAdjacencyList("""
1 C u0 p0 c0 i13 {2,S} {3,S} {4,S} {5,S}
2 H u0 p0 c0 {1,S}
3 H u0 p0 c0 {1,S}
4 H u0 p0 c0 {1,S}
5 H u0 p0 c0 {1,S}
""")
methane = Molecule().fromAdjacencyList("""
1 C u0 p0 c0 {2,S} {3,S} {4,S} {5,S}
2 H u0 p0 c0 {1,S}
3 H u0 p0 c0 {1,S}
4 H u0 p0 c0 {1,S}
5 H u0 p0 c0 {1,S}
""")
group_methane = Group().fromAdjacencyList("""
1 C u0 {2,S} {3,S} {4,S} {5,S}
2 H u0 {1,S}
3 H u0 {1,S}
4 H u0 {1,S}
5 H u0 {1,S}
""")
assert_true(methanei.isSubgraphIsomorphic(group_methane))
assert_true(methane.isSubgraphIsomorphic(group_methane))
assert_false(methanei.isIsomorphic(methane))
def test_isomorphism_sulfurGroup_sulfurMolecule():
"""
Test isormophism check of a CS group vs. a sulfur containing molecule
"""
gp1 = Group().fromAdjacencyList("""
1 S u0 {2,S} {3,S}
2 H u0 {1,S}
3 C u0 {1,S} {4,D}
4 S u0 {3,D}
""")
gp2 = Group().fromAdjacencyList("""
1 S u0 {2,S} {3,S}
2 H u0 {1,S}
3 CS u0 {1,S} {4,D}
4 S u0 {3,D}
""")
mol = Molecule().fromAdjacencyList("""
1 S 0 {2,S} {3,S}
2 H 0 {1,S}
3 C 0 {1,S} {4,D} {5,S}
4 S 0 {3,D}
5 H 0 {3,S}
""")
assert_true(mol.isSubgraphIsomorphic(gp1))
assert_true(mol.isSubgraphIsomorphic(gp2))
def test_isomorphism_sulfurGroup_sulfurMolecule():
"""
Test isormophism check of a CS group vs. a sulfur containing molecule
"""
gp1 = Group().fromAdjacencyList("""
1 S u0 {2,S} {3,S}
2 H u0 {1,S}
3 C u0 {1,S} {4,D}
4 S u0 {3,D}
""")
gp2 = Group().fromAdjacencyList("""
1 S u0 {2,S} {3,S}
2 H u0 {1,S}
3 CS u0 {1,S} {4,D}
4 S u0 {3,D}
""")
mol = Molecule().fromAdjacencyList("""
1 S 0 {2,S} {3,S}
2 H 0 {1,S}
3 C 0 {1,S} {4,D} {5,S}
4 S 0 {3,D}
5 H 0 {3,S}
""")
assert_true(mol.isSubgraphIsomorphic(gp1))
assert_true(mol.isSubgraphIsomorphic(gp2))
def testSubgraphIsomorphismManyLabels(self):
molecule = Molecule() # specific case (species)
molecule.fromAdjacencyList("""
1 *1 C u1 p0 c0 {2,S} {3,S} {4,S}
2 C u0 p0 c0 {1,S} {3,S} {5,S} {6,S}
3 C u0 p0 c0 {1,S} {2,S} {7,S} {8,S}
4 H u0 p0 c0 {1,S}
5 H u0 p0 c0 {2,S}
6 H u0 p0 c0 {2,S}
7 H u0 p0 c0 {3,S}
8 H u0 p0 c0 {3,S}
""")
group = Group() # general case (functional group)
group.fromAdjacencyList("""
1 *1 C u1 p0 c0 {2,S}, {3,S}
2 R!H u0 p0 c0 {1,S}
3 R!H u0 p0 c0 {1,S}
""")
labeled1 = molecule.getLabeledAtoms()
labeled2 = group.getLabeledAtoms()
initialMap = {}
for label,atom1 in labeled1.iteritems():
initialMap[atom1] = labeled2[label]
self.assertTrue(molecule.isSubgraphIsomorphic(group, initialMap))
mapping = molecule.findSubgraphIsomorphisms(group, initialMap)
self.assertEqual(len(mapping), 2)
for map in mapping:
self.assertTrue(len(map) == min(len(molecule.atoms), len(group.atoms)))
for key, value in map.iteritems():
self.assertTrue(key in molecule.atoms)
self.assertTrue(value in group.atoms)
def testSubgraphIsomorphismManyLabels(self):
molecule = Molecule() # specific case (species)
molecule.fromAdjacencyList("""
1 *1 C u1 p0 c0 {2,S} {3,S} {4,S}
2 C u0 p0 c0 {1,S} {3,S} {5,S} {6,S}
3 C u0 p0 c0 {1,S} {2,S} {7,S} {8,S}
4 H u0 p0 c0 {1,S}
5 H u0 p0 c0 {2,S}
6 H u0 p0 c0 {2,S}
7 H u0 p0 c0 {3,S}
8 H u0 p0 c0 {3,S}
""")
group = Group() # general case (functional group)
group.fromAdjacencyList("""
1 *1 C u1 p0 c0 {2,S}, {3,S}
2 R!H u0 p0 c0 {1,S}
3 R!H u0 p0 c0 {1,S}
""")
labeled1 = molecule.getLabeledAtoms()
labeled2 = group.getLabeledAtoms()
initialMap = {}
for label, atom1 in labeled1.iteritems():
initialMap[atom1] = labeled2[label]
self.assertTrue(molecule.isSubgraphIsomorphic(group, initialMap))
mapping = molecule.findSubgraphIsomorphisms(group, initialMap)
self.assertEqual(len(mapping), 2)
for map in mapping:
self.assertTrue(
len(map) == min(len(molecule.atoms), len(group.atoms)))
for key, value in map.iteritems():
self.assertTrue(key in molecule.atoms)
self.assertTrue(value in group.atoms)
def test_isomorphism_mol_group_not_identical():
"""
Testing multiplicities in mol and group that don't match
"""
mol = Molecule().fromAdjacencyList("""
1 C u0 p0 c0
""", saturateH=True)
gp = Group().fromAdjacencyList("""
multiplicity [2]
1 R u0 p0 c0
""")
assert_false(mol.isSubgraphIsomorphic(gp))
assert_false(len(mol.findSubgraphIsomorphisms(gp)) > 0)
def test_isomorphism_mol_group_not_identical():
"""
Testing multiplicities in mol and group that don't match
"""
mol = Molecule().fromAdjacencyList("""
1 C u0 p0 c0
""", saturateH=True)
gp = Group().fromAdjacencyList("""
multiplicity [2]
1 R u0 p0 c0
""")
assert_false(mol.isSubgraphIsomorphic(gp))
assert_false(len(mol.findSubgraphIsomorphisms(gp)) > 0)
def testSubgraphIsomorphismAgain(self):
molecule = Molecule()
molecule.fromAdjacencyList("""
1 * C u0 p0 c0 {2,D} {7,S} {8,S}
2 C u0 p0 c0 {1,D} {3,S} {9,S}
3 C u0 p0 c0 {2,S} {4,D} {10,S}
4 C u0 p0 c0 {3,D} {5,S} {11,S}
5 C u0 p0 c0 {4,S} {6,S} {12,S} {13,S}
6 C u0 p0 c0 {5,S} {14,S} {15,S} {16,S}
7 H u0 p0 c0 {1,S}
8 H u0 p0 c0 {1,S}
9 H u0 p0 c0 {2,S}
10 H u0 p0 c0 {3,S}
11 H u0 p0 c0 {4,S}
12 H u0 p0 c0 {5,S}
13 H u0 p0 c0 {5,S}
14 H u0 p0 c0 {6,S}
15 H u0 p0 c0 {6,S}
16 H u0 p0 c0 {6,S}
""")
group = Group()
group.fromAdjacencyList("""
1 * C u0 p0 c0 {2,D} {3,S} {4,S}
2 C u0 p0 c0 {1,D}
3 H u0 p0 c0 {1,S}
4 H u0 p0 c0 {1,S}
""")
labeled1 = molecule.getLabeledAtoms().values()[0]
labeled2 = group.getLabeledAtoms().values()[0]
initialMap = {labeled1: labeled2}
self.assertTrue(molecule.isSubgraphIsomorphic(group, initialMap))
initialMap = {labeled1: labeled2}
mapping = molecule.findSubgraphIsomorphisms(group, initialMap)
self.assertTrue(
len(mapping) == 2,
"len(mapping) = %d, should be = 2" % (len(mapping)))
for map in mapping:
self.assertTrue(
len(map) == min(len(molecule.atoms), len(group.atoms)))
for key, value in map.iteritems():
self.assertTrue(key in molecule.atoms)
self.assertTrue(value in group.atoms)
def testSubgraphIsomorphism(self):
"""
Check the graph isomorphism functions.
"""
molecule = Molecule().fromSMILES('C=CC=C[CH]C')
group = Group().fromAdjacencyList("""
1 Cd u0 p0 c0 {2,D}
2 Cd u0 p0 c0 {1,D}
""")
self.assertTrue(molecule.isSubgraphIsomorphic(group))
mapping = molecule.findSubgraphIsomorphisms(group)
self.assertTrue(len(mapping) == 4, "len(mapping) = %d, should be = 4" % (len(mapping)))
for map in mapping:
self.assertTrue(len(map) == min(len(molecule.atoms), len(group.atoms)))
for key, value in map.iteritems():
self.assertTrue(key in molecule.atoms)
self.assertTrue(value in group.atoms)
def testSubgraphIsomorphismAgain(self):
molecule = Molecule()
molecule.fromAdjacencyList("""
1 * C u0 p0 c0 {2,D} {7,S} {8,S}
2 C u0 p0 c0 {1,D} {3,S} {9,S}
3 C u0 p0 c0 {2,S} {4,D} {10,S}
4 C u0 p0 c0 {3,D} {5,S} {11,S}
5 C u0 p0 c0 {4,S} {6,S} {12,S} {13,S}
6 C u0 p0 c0 {5,S} {14,S} {15,S} {16,S}
7 H u0 p0 c0 {1,S}
8 H u0 p0 c0 {1,S}
9 H u0 p0 c0 {2,S}
10 H u0 p0 c0 {3,S}
11 H u0 p0 c0 {4,S}
12 H u0 p0 c0 {5,S}
13 H u0 p0 c0 {5,S}
14 H u0 p0 c0 {6,S}
15 H u0 p0 c0 {6,S}
16 H u0 p0 c0 {6,S}
""")
group = Group()
group.fromAdjacencyList("""
1 * C u0 p0 c0 {2,D} {3,S} {4,S}
2 C u0 p0 c0 {1,D}
3 H u0 p0 c0 {1,S}
4 H u0 p0 c0 {1,S}
""")
labeled1 = molecule.getLabeledAtoms().values()[0]
labeled2 = group.getLabeledAtoms().values()[0]
initialMap = {labeled1: labeled2}
self.assertTrue(molecule.isSubgraphIsomorphic(group, initialMap))
initialMap = {labeled1: labeled2}
mapping = molecule.findSubgraphIsomorphisms(group, initialMap)
self.assertTrue(len(mapping) == 2, "len(mapping) = %d, should be = 2" % (len(mapping)))
for map in mapping:
self.assertTrue(len(map) == min(len(molecule.atoms), len(group.atoms)))
for key, value in map.iteritems():
self.assertTrue(key in molecule.atoms)
self.assertTrue(value in group.atoms)
def testSubgraphIsomorphism(self):
"""
Check the graph isomorphism functions.
"""
molecule = Molecule().fromSMILES('C=CC=C[CH]C')
group = Group().fromAdjacencyList("""
1 Cd u0 p0 c0 {2,D}
2 Cd u0 p0 c0 {1,D}
""")
self.assertTrue(molecule.isSubgraphIsomorphic(group))
mapping = molecule.findSubgraphIsomorphisms(group)
self.assertTrue(
len(mapping) == 4,
"len(mapping) = %d, should be = 4" % (len(mapping)))
for map in mapping:
self.assertTrue(
len(map) == min(len(molecule.atoms), len(group.atoms)))
for key, value in map.iteritems():
self.assertTrue(key in molecule.atoms)
self.assertTrue(value in group.atoms)
