File: test2.cpp

package info (click to toggle)
rdkit 201809.1%2Bdfsg-6
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 123,688 kB
  • sloc: cpp: 230,509; python: 70,501; java: 6,329; ansic: 5,427; sql: 1,899; yacc: 1,739; lex: 1,243; makefile: 445; xml: 229; fortran: 183; sh: 123; cs: 93
file content (116 lines) | stat: -rw-r--r-- 3,240 bytes parent folder | download | duplicates (3)
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
113
114
115
116
//
//  Copyright (C) 2018 Greg Landrum
//
//   @@ 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 <RDGeneral/test.h>
#include <iostream>
#include <string>
#include <GraphMol/RDKitBase.h>
#include "SmilesParse.h"
#include "SmilesWrite.h"
#include "SmartsWrite.h"
#include <RDGeneral/RDLog.h>
//#include <boost/log/functions.hpp>
using namespace RDKit;
using namespace std;

void testParseAtomSmiles() {
  BOOST_LOG(rdInfoLog) << "Testing ParseAtomSmiles" << std::endl;
  {  // these should pass
    std::vector<std::string> smiles = {"C", "[NH4+]", "[13C@H]"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Atom> a1(SmilesToAtom(pr));
      TEST_ASSERT(a1);
      TEST_ASSERT(a1->getAtomicNum() > 0);
    }
  }

  {  // these should fail
    std::vector<std::string> smiles = {"CO", "", "C-O", "-", "[Bg]"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Atom> a1(SmilesToAtom(pr));
      TEST_ASSERT(!a1);
    }
  }
  BOOST_LOG(rdInfoLog) << "done" << std::endl;
}

void testParseBondSmiles() {
  BOOST_LOG(rdInfoLog) << "Testing ParseBondSmiles" << std::endl;
  {  // these should pass
    std::vector<std::string> smiles = {":", "=", "#", "-", "/", "\\"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Bond> a1(SmilesToBond(pr));
      TEST_ASSERT(a1);
    }
  }

  {  // these should fail
    std::vector<std::string> smiles = {"C", "", "C-O", "*"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Bond> a1(SmilesToBond(pr));
      TEST_ASSERT(!a1);
    }
  }
  BOOST_LOG(rdInfoLog) << "done" << std::endl;
}

void testParseAtomSmarts() {
  BOOST_LOG(rdInfoLog) << "Testing ParseAtomSmarts" << std::endl;
  {  // these should pass
    std::vector<std::string> smiles = {"C", "[NH4+]", "[13C@H]",
                                       "[C,N,$(C=CC)]"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Atom> a1(SmartsToAtom(pr));
      TEST_ASSERT(a1);
    }
  }

  {  // these should fail
    std::vector<std::string> smiles = {"CO", "", "C-O", "-", "[Bg]"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Atom> a1(SmartsToAtom(pr));
      TEST_ASSERT(!a1);
    }
  }
  BOOST_LOG(rdInfoLog) << "done" << std::endl;
}

void testParseBondSmarts() {
  BOOST_LOG(rdInfoLog) << "Testing ParseBondSmarts" << std::endl;
  {  // these should pass
    std::vector<std::string> smiles = {":",  "=", "#",  "-",  "/",
                                       "\\", "@", "!-", "=,#"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Bond> a1(SmartsToBond(pr));
      TEST_ASSERT(a1);
    }
  }

  {  // these should fail
    std::vector<std::string> smiles = {"C", "", "-O", "*"};
    for (const auto &pr : smiles) {
      std::unique_ptr<Bond> a1(SmartsToBond(pr));
      TEST_ASSERT(!a1);
    }
  }
  BOOST_LOG(rdInfoLog) << "done" << std::endl;
}

int main(int argc, char *argv[]) {
  (void)argc;
  (void)argv;
  RDLog::InitLogs();
// boost::logging::enable_logs("rdApp.debug");
#if 1
  testParseAtomSmiles();
  testParseBondSmiles();
  testParseAtomSmarts();
  testParseBondSmarts();
#endif
}