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
|
//
// Copyright (C) 2021 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 "catch.hpp"
#include <GraphMol/RDKitBase.h>
#include <GraphMol/SmilesParse/SmilesParse.h>
#include <GraphMol/SmilesParse/SmilesWrite.h>
#include <algorithm>
using namespace RDKit;
TEST_CASE("mol.atoms()") {
const auto m = "CC(C)CO"_smiles;
REQUIRE(m);
unsigned int ccount = 0;
for (const auto atom : m->atoms()) {
if (atom->getAtomicNum() == 6) {
++ccount;
}
}
CHECK(ccount == 4);
}
TEST_CASE("mol.atomNeighbors()") {
const auto m = "CC(C)CO"_smiles;
REQUIRE(m);
unsigned int count = 0;
for (const auto atom : m->atomNeighbors(m->getAtomWithIdx(1))) {
count += atom->getDegree();
}
CHECK(count == 4);
for (auto atom : m->atomNeighbors(m->getAtomWithIdx(1))) {
atom->setAtomicNum(7);
}
MolOps::sanitizeMol(*m);
CHECK(MolToSmiles(*m) == "NC(N)NO");
}
TEST_CASE("mol.atomBonds()") {
const auto m = "CC(=C)CO"_smiles;
REQUIRE(m);
double count = 0;
for (const auto bond : m->atomBonds(m->getAtomWithIdx(1))) {
count += bond->getBondTypeAsDouble();
}
CHECK(count == 4);
for (auto bond : m->atomBonds(m->getAtomWithIdx(1))) {
bond->setBondType(Bond::BondType::SINGLE);
}
MolOps::sanitizeMol(*m);
CHECK(MolToSmiles(*m) == "CC(C)CO");
}
|