1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#
# Created by Gareth Jones on 5/30/2020.
#
# Copyright 2020-2022 Schrodinger, Inc and other RDKit contributors
# @@ All Rights Reserved @@
# This file is part of the RDKit.
# The contents are covered by the terms of the BSD license
# which is included in the file license.txt, found at the root
# of the RDKit source tree.
""" A rough test of Tautomer Queries
Rough in that only basic functionality is evaluated.
"""
import os
import pickle
from unittest import TestCase, main
import unittest
from rdkit import Chem, DataStructs
from rdkit.Chem import rdTautomerQuery
from rdkit import rdBase
class TautomerQueryTestCase(TestCase):
def test_basic(self):
mol = Chem.MolFromSmiles("O=C1CCCCC1")
tautomer_query = rdTautomerQuery.TautomerQuery(mol)
self.assertEqual(2, len(tautomer_query.GetTautomers()))
modified_bonds = tautomer_query.GetModifiedBonds()
self.assertEqual(3, len(modified_bonds))
modified_atoms = tautomer_query.GetModifiedAtoms()
self.assertEqual(3, len(modified_atoms))
target = Chem.MolFromSmiles("OC1=CCCC(CC)C1")
self.assertTrue(target.HasSubstructMatch(tautomer_query.GetTemplateMolecule()))
match = tautomer_query.IsSubstructOf(target)
self.assertTrue(match)
self.assertTrue(target.HasSubstructMatch(tautomer_query.GetTemplateMolecule()))
#self.assertTrue(target.HasSubstructMatch(tautomer_query.GetTautomers()[1]))
match = tautomer_query.GetSubstructMatch(target)
self.assertEqual(7, len(match))
matches = tautomer_query.GetSubstructMatches(target)
self.assertEqual(1, len(matches))
tautomer_matches = tautomer_query.GetSubstructMatchesWithTautomers(target)
self.assertEqual(1, len(tautomer_matches))
self.assertEqual(len(tautomer_matches[0][0]), 7)
matching_tautomer = tautomer_matches[0][1]
self.assertTrue(target.HasSubstructMatch(matching_tautomer))
tautomer_fingerprint = tautomer_query.PatternFingerprintTemplate()
target_fingerprint = rdTautomerQuery.PatternFingerprintTautomerTarget(target)
matching_fingerprints = DataStructs.AllProbeBitsMatch(tautomer_fingerprint, target_fingerprint)
self.assertTrue(matching_fingerprints)
file = os.environ['RDBASE'] + '/Code/GraphMol/MolStandardize/test_data/tautomerTransforms.in'
tautomer_query2 = rdTautomerQuery.TautomerQuery(mol, file)
self.assertEqual(2, len(tautomer_query.GetTautomers()))
def test_parameter_searches(self):
mol = Chem.MolFromSmiles("O=C1CCCCC1")
tautomer_query = rdTautomerQuery.TautomerQuery(mol)
target = Chem.MolFromSmiles("O=C1CCCC(CC)C1")
params = Chem.SubstructMatchParameters()
params.uniquify = False
tautomer_matches = tautomer_query.GetSubstructMatchesWithTautomers(target, params)
self.assertEqual(2, len(tautomer_matches))
matches = tautomer_query.GetSubstructMatches(target, params)
self.assertEqual(2, len(matches))
match = tautomer_query.GetSubstructMatch(target, params)
self.assertEqual(7, len(match))
is_match = tautomer_query.IsSubstructOf(target, params)
self.assertTrue(is_match)
def test_types(self):
mol = Chem.MolFromSmiles("O=C1CCCCC1")
tautomer_query = rdTautomerQuery.TautomerQuery(mol)
target = Chem.MolFromSmiles("OC1=CCCC(CC)C1")
try:
self.assertTrue(target.HasSubstructMatch(tautomer_query.GetTautomers()[1]))
except Exception:
self.fail("Boost type error")
def test_simple(self):
mol = Chem.MolFromSmiles("CC=O")
tautomer_query = rdTautomerQuery.TautomerQuery(mol)
target = Chem.MolFromSmiles("OC(C)=O")
tautomers = tautomer_query.GetTautomers()
self.assertTrue(target.HasSubstructMatch(mol))
self.assertTrue(tautomer_query.IsSubstructOf(target))
@unittest.skipIf(not rdBase._serializationEnabled, "not built with serialization support")
def test_serialization(self):
mol = Chem.MolFromSmiles("O=C1CCCCC1")
base_tautomer_query = rdTautomerQuery.TautomerQuery(mol)
for tautomer_query in (pickle.loads(pickle.dumps(base_tautomer_query)),
rdTautomerQuery.TautomerQuery(base_tautomer_query.ToBinary())):
self.assertEqual(2, len(tautomer_query.GetTautomers()))
modified_bonds = tautomer_query.GetModifiedBonds()
self.assertEqual(3, len(modified_bonds))
modified_atoms = tautomer_query.GetModifiedAtoms()
self.assertEqual(3, len(modified_atoms))
target = Chem.MolFromSmiles("OC1=CCCC(CC)C1")
self.assertTrue(target.HasSubstructMatch(tautomer_query.GetTemplateMolecule()))
if __name__ == "__main__":
main()
|
