Add meaningful tests to new-layout

python-poetry · Apr 29, 2020 · cfe6865 · cfe6865
1 parent 5466b97
commit cfe6865
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git 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
@@ -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
@@ -5,11 +5,18 @@
 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
 """