diff --git a/Tests/test_functions.py b/Tests/test_functions.py index d98eef1..0405526 100644 --- a/Tests/test_functions.py +++ b/Tests/test_functions.py @@ -1,8 +1,9 @@ import pytest from DECIMER import predict_SMILES -def get_imagetosmiles(): +@pytest.mark.filterwarnings('ignore::DeprecationWarning') +def test_imagetosmiles(): img_path = 'Tests/caffeine.png' expected_result = "CN1C=NC2=C1C(=O)N(C(=O)N2C)C" - actual_result = predict_SMILES(image_path) + actual_result = predict_SMILES(img_path) assert expected_result == actual_result \ No newline at end of file diff --git a/setup.py b/setup.py index 033c79d..2a6c65d 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ packages=setuptools.find_packages(), license="MIT", install_requires=[ - "tensorflow==2.10.0", + "tensorflow==2.10.1", "opencv-python", "pystow", "pyheif", diff --git a/tox.ini b/tox.ini index 7f936ba..7585099 100644 --- a/tox.ini +++ b/tox.ini @@ -1,5 +1,5 @@ [tox] -envlist = py38, lint +envlist = py310, lint requires = tox-conda [testenv] @@ -10,7 +10,7 @@ conda_deps = pytest conda_channels = conda-forge -commands = pytest --basetemp="{envtmpdir}" {posargs} +commands = pytest -p no:warnings --basetemp="{envtmpdir}" {posargs} [testenv:lint] basepython = python3 @@ -42,3 +42,4 @@ exclude = DECIMER/efficientnetv2/hparams.py, DECIMER/efficientnetv2/utils.py, Tests/test_functions.py, + \ No newline at end of file