test: Fix imports for unit/integration tests

Author: jmgate

test/__init__.py

@@ -0,0 +1,7 @@
+"""
+Create the ``test`` package.
+
+This ``__init__.py`` file creates the ``test`` package, such that tests
+can relative-import from modules in the sibling ``staged_script``
+directory.
+"""

test/test_staged_script.py

@@ -8,10 +8,7 @@
 import pytest
 from rich.console import Console
 
-from python.staged_script.staged_script.staged_script import (
-    StagedScript,
-    StageDuration,
-)
+from staged_script import StagedScript, StageDuration
 
 
 @pytest.fixture()

test/test_staged_script_advanced_subclass.py

@@ -6,7 +6,7 @@
 from rich.console import Console
 from tenacity import RetryCallState, Retrying, TryAgain
 
-from python.staged_script.staged_script.staged_script import StagedScript
+from staged_script import StagedScript
 
 
 class MyAdvancedScript(StagedScript):

test/test_staged_script_basic_subclass.py

@@ -3,7 +3,7 @@
 import pytest
 from rich.console import Console
 
-from python.staged_script.staged_script.staged_script import StagedScript
+from staged_script import StagedScript
 
 
 class MyBasicScript(StagedScript):

test/test_staged_script_registered_stages.py

@@ -1,6 +1,6 @@
 """Integration tests for retry options."""
 
-from python.staged_script.staged_script.staged_script import StagedScript
+from staged_script import StagedScript
 
 
 class MyScript(StagedScript):