Add the correct path when it's a package

Author: Michael Fraenkel

integration/_support/package/tasks/module.py

@@ -1,4 +1,6 @@
 from invoke import task
+from . import pytest as pt
+from pytest import Testdir
 
 
 @task

integration/_support/package/tasks/pytest.py

@@ -73,6 +73,8 @@ def load(self, name: Optional[str] = None) -> Tuple[ModuleType, str]:
             # being imported is trying to load local-to-it names.
             if os.path.isfile(spec.origin):
                 path = os.path.dirname(spec.origin)
+                if spec.origin.endswith("__init__.py"):
+                    path = os.path.dirname(path)
             if path not in sys.path:
                 sys.path.insert(0, path)
             # Actual import

invoke/loader.py

@@ -51,6 +51,12 @@ def adds_module_parent_dir_to_sys_path(self):
         # Crummy doesn't-explode test.
         _BasicLoader().load("namespacing")
 
+    def adds_package_dir_to_sys_path(self):
+        config = Config({"tasks": {"collection_name": "module"}})
+        _BasicLoader(config).load("package")
+        package = Path(support) / "package"
+        assert str(package) not in sys.path
+
     def doesnt_duplicate_parent_dir_addition(self):
         _BasicLoader().load("namespacing")
         _BasicLoader().load("namespacing")