mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-07 22:44:25 +08:00
f04a9c9a42
- switched to using std::vector<int> instead of std::list<int> for _ErGAtomTypes property - added RegisterListConverter<std::vector<unsigned int>> Co-authored-by: Paolo Tosco <paolo.tosco@novartis.com>
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
# $Id$
|
|
#
|
|
from rdkit import Chem
|
|
from rdkit.Chem import rdReducedGraphs as rdRG
|
|
from rdkit import RDConfig
|
|
import numpy
|
|
import unittest
|
|
|
|
|
|
class TestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def test1(self):
|
|
m = Chem.MolFromSmiles('OCCc1ccccc1')
|
|
mrg = rdRG.GenerateMolExtendedReducedGraph(m)
|
|
mrg.UpdatePropertyCache(False)
|
|
self.assertEqual('*cCCO', Chem.MolToSmiles(mrg))
|
|
|
|
m = Chem.MolFromSmiles('OCCC1CCCCC1')
|
|
mrg = rdRG.GenerateMolExtendedReducedGraph(m)
|
|
mrg.UpdatePropertyCache(False)
|
|
self.assertEqual('*CCCO', Chem.MolToSmiles(mrg))
|
|
|
|
def test2(self):
|
|
m = Chem.MolFromSmiles('OCCc1ccccc1')
|
|
mrg = rdRG.GenerateMolExtendedReducedGraph(m)
|
|
mrg.UpdatePropertyCache(False)
|
|
self.assertEqual('*cCCO', Chem.MolToSmiles(mrg))
|
|
|
|
fp1 = rdRG.GenerateErGFingerprintForReducedGraph(mrg)
|
|
fp2 = rdRG.GetErGFingerprint(m)
|
|
md = max(abs(fp1 - fp2))
|
|
self.assertLess(md, 1e-4)
|
|
|
|
def test3(self):
|
|
m = Chem.MolFromSmiles('OCCc1ccccc1')
|
|
fp1 = rdRG.GetErGFingerprint(m)
|
|
m = Chem.MolFromSmiles('OCCC1CC=CC=C1')
|
|
fp2 = rdRG.GetErGFingerprint(m)
|
|
|
|
md = max(abs(fp1 - fp2))
|
|
self.assertAlmostEqual(0.0, md, 4)
|
|
|
|
def test4(self):
|
|
m = Chem.MolFromSmiles('OCCc1ccccc1')
|
|
fp1 = rdRG.GetErGFingerprint(m)
|
|
fp2 = rdRG.GetErGFingerprint(m, fuzzIncrement=0.1)
|
|
|
|
md = max(abs(fp1 - fp2))
|
|
self.assertAlmostEqual(0.2, md, 4)
|
|
|
|
def testCanRetrieveProp(self):
|
|
m = Chem.MolFromSmiles('OCCc1ccccc1')
|
|
mrg = rdRG.GenerateMolExtendedReducedGraph(m)
|
|
erg_types = [tuple(atom.GetPropsAsDict().get('_ErGAtomTypes')) for atom in mrg.GetAtoms()]
|
|
self.assertEqual(erg_types, [(0, 1), (), (), (), (5,)])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|