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)