diff --git a/pystencils/__init__.py b/pystencils/__init__.py index c74a03e175a711e8082ad03644f549d9c5d73d3f..d2abab709fda4c8ceb4e198b364ea480a45a3e1c 100644 --- a/pystencils/__init__.py +++ b/pystencils/__init__.py @@ -22,16 +22,16 @@ except ImportError: pass -def _get_file_path(): +def _get_release_file(): import os.path file_path = os.path.abspath(os.path.dirname(__file__)) return os.path.join(file_path, '..', 'RELEASE-VERSION') try: - __version__ = open(_get_file_path(), 'r').read() + __version__ = open(_get_release_file(), 'r').read() except IOError: - __version__ = 'Please install pystencils with setup.py' + __version__ = 'development' __all__ = ['Field', 'FieldType', 'fields', 'TypedSymbol', diff --git a/pystencils_tests/test_version_string.py b/pystencils_tests/test_version_string.py new file mode 100644 index 0000000000000000000000000000000000000000..541ba35e2435f03b2ef9ef17054de2aa14c172fa --- /dev/null +++ b/pystencils_tests/test_version_string.py @@ -0,0 +1,12 @@ +import pystencils as ps +from pathlib import Path + +def test_version_string(): + file_path = Path(__file__).parent + release_version = file_path.parent.absolute() / 'RELEASE-VERSION' + if release_version.exists (): + with open(release_version, "r") as f: + version = f.read() + assert ps.__version__ == version + else: + ps.__version__ == "development"