diff --git a/docs/docs/basic-usage.md b/docs/docs/basic-usage.md index dad1ef9b9b4..6ce54a9b3b0 100644 --- a/docs/docs/basic-usage.md +++ b/docs/docs/basic-usage.md @@ -39,6 +39,7 @@ python = "*" [tool.poetry.dev-dependencies] pytest = "^3.4" +toml = "^0.10.0" ``` ### Specifying dependencies diff --git a/poetry/console/commands/new.py b/poetry/console/commands/new.py index 53e61702f90..2f4b12aa175 100644 --- a/poetry/console/commands/new.py +++ b/poetry/console/commands/new.py @@ -59,7 +59,9 @@ def handle(self): ".".join(str(v) for v in current_env.version_info[:2]) ) - dev_dependencies = {} + dev_dependencies = { + "toml": "^0.10.0", + } python_constraint = parse_constraint(default_python) if parse_constraint("<3.5").allows_any(python_constraint): dev_dependencies["pytest"] = "^4.6" diff --git a/poetry/layouts/layout.py b/poetry/layouts/layout.py index a7378c67c4c..40c5e8a213e 100644 --- a/poetry/layouts/layout.py +++ b/poetry/layouts/layout.py @@ -5,11 +5,20 @@ from poetry.utils.helpers import module_name -TESTS_DEFAULT = u"""from {package_name} import __version__ +TESTS_DEFAULT = u"""\ +from pathlib import Path + +import toml + +from {package_name} import __version__ as module_version def test_version(): - assert __version__ == '{version}' + pyproject_file = Path(__file__).parents[1] / "pyproject.toml" + pyproject = toml.load(pyproject_file) + pyproject_version = pyproject["tool"]["poetry"]["version"] + + assert module_version == pyproject_version """