File: connectTheDots_catch.cpp

package info (click to toggle)
rdkit 202503.1-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 220,160 kB
  • sloc: cpp: 399,240; python: 77,453; ansic: 25,517; java: 8,173; javascript: 4,005; sql: 2,389; yacc: 1,565; lex: 1,263; cs: 1,081; makefile: 580; xml: 229; fortran: 183; sh: 105
file content (66 lines) | stat: -rw-r--r-- 2,247 bytes parent folder | download | duplicates (2)
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
//
//  Copyright (C) 2023 Greg Landrum 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.
//

#include <algorithm>
#include <fstream>
#include <string>
#include <sstream>
#include <string_view>
#include <streambuf>

#include "RDGeneral/test.h"
#include <catch2/catch_all.hpp>
#include <RDGeneral/Invariant.h>
#include <GraphMol/RDKitBase.h>
#include <GraphMol/FileParsers/FileParsers.h>
#include <GraphMol/SmilesParse/SmilesParse.h>
#include <GraphMol/FileParsers/ProximityBonds.h>

using namespace RDKit;

TEST_CASE(
    "Github #6756: ConnectTheDots can segfault if atoms do not have residue info") {
  SECTION("small molecule, was not a problem") {
    std::string smiles =
        "[H]C([H])=C([H])C([H])([H])[H] |(1.35283,-0.241953,-1.28693;1.35436,-0.216135,-0.201646;2.30673,-0.352635,0.30154;0.23076,-0.0239947,0.500064;0.281106,-0.00576639,1.58653;-1.11716,0.171588,-0.112486;-1.08265,0.139951,-1.20609;-1.52627,1.14083,0.188471;-1.79971,-0.611884,0.230548)|";
    SmilesParserParams ps;
    ps.removeHs = false;
    std::unique_ptr<RWMol> m(SmilesToMol(smiles, ps));
    REQUIRE(m);
    CHECK(m->getNumBonds() == 8);
    // start without bonds
    m->beginBatchEdit();
    for (auto bond : m->bonds()) {
      m->removeBond(bond->getBeginAtomIdx(), bond->getEndAtomIdx());
    }
    m->commitBatchEdit();
    CHECK(m->getNumBonds() == 0);
    ConnectTheDots(m.get());
    CHECK(m->getNumBonds() == 8);
  }
  SECTION("small molecule with bivalent H, this did crash") {
    std::string smiles = "F[H]F |(1.1,0,0;0,0,0;-1.1,0,0)|";
    SmilesParserParams ps;
    ps.removeHs = false;
    ps.sanitize = false;
    std::unique_ptr<RWMol> m(SmilesToMol(smiles, ps));
    REQUIRE(m);
    m->updatePropertyCache(false);
    CHECK(m->getNumBonds() == 2);
    // start without bonds
    m->beginBatchEdit();
    for (auto bond : m->bonds()) {
      m->removeBond(bond->getBeginAtomIdx(), bond->getEndAtomIdx());
    }
    m->commitBatchEdit();
    CHECK(m->getNumBonds() == 0);
    ConnectTheDots(m.get());
    CHECK(m->getNumBonds() == 1);
  }
}