def test_database_query():
with assert_raises(ValueError):
queries.advanced_search(databases.MINE('admin'), "{'MINE_id': 19160}")
with assert_raises(ValueError):
queries.advanced_search(test_db, "")
assert queries.advanced_search(test_db, "{'MINE_id': 917030}") == [glucose]
assert queries.advanced_search(test_db,
"{'Names': 'Glucose'}") == [glucose]
assert queries.advanced_search(test_db, "{'MINE_id': 917030}",
{'_id': 1}) == [glucose_id]
def test_database_query(test_db, glucose, glucose_id):
"""
GIVEN an andvanced search query (e.g. a MINE id)
WHEN advanced search is used to search based on that query
THEN make sure that advanced search provides the correct results
"""
with pytest.raises(ValueError):
queries.advanced_search(databases.MINE('admin'), "{'MINE_id': 19160}")
with pytest.raises(ValueError):
queries.advanced_search(test_db, "")
assert queries.advanced_search(test_db, "{'MINE_id': 917030}") == [glucose]
assert queries.advanced_search(test_db,
"{'Names': 'Glucose'}") == [glucose]
assert queries.advanced_search(test_db, "{'MINE_id': 917030}",
{'_id': 1}) == [glucose_id]
import pymongo
import unittest
from minedatabase import databases
from minedatabase import utils
from rdkit.Chem import AllChem
import os
from shutil import rmtree
test_db = databases.MINE('mongotest')
@unittest.skipIf("TRAVIS" in os.environ and os.environ["TRAVIS"] == "true",
"Skipping this test on Travis CI.")
def test_generate_image_files():
img_dir = os.getcwd() + '/imgs'
test_db.generate_image_files(img_dir)
try:
assert os.path.exists(
os.path.join(img_dir,
'./C455bc3dc93cd3bb3bef92a34767693a4716aa3fb.svg'))
print(len(os.listdir(img_dir)))
assert len(os.listdir(img_dir)) == 26
finally:
rmtree(img_dir)
test_db.generate_image_files(img_dir, {'Generation': 1}, 3, 'png')
try:
assert os.path.exists(
os.path.join(img_dir, 'C', 'c', 'f',
'Ccffda1b2e82fcdb0e1e710cad4d5f70df7a5d74f.png'))
finally:
rmtree(img_dir)