Add the correct path when it's a package

Author: Michael Fraenkel

integration/_support/package/tasks/module.py

@@ -1,6 +1,10 @@
 from invoke import task
+from . import pytest as pt
+import pytest
+
+pytest.__version__
 
 
 @task
 def mytask(c):
-    print("hi!")
+    print(pt.hi)

integration/_support/package/tasks/pytest.py

@@ -0,0 +1 @@
+hi = "hi!"

invoke/loader.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

tests/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")