Update linting and tests.

TGSAI · Nov 3, 2023 · 40f32f8 · 40f32f8
1 parent 864a3b3
commit 40f32f8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/mdio/__main__.py b/src/mdio/__main__.py
@@ -10,8 +10,9 @@
 import click_params
 
 import mdio
+from mdio.commands.utility import copy
+from mdio.commands.utility import info
 
-from mdio.commands.utility import copy, info
 
 plugin_folder = os.path.join(os.path.dirname(__file__), "commands")
 

diff --git a/src/mdio/commands/utility.py b/src/mdio/commands/utility.py
@@ -1,11 +1,12 @@
 """Dataset CLI Plugin."""
 
 try:
+    from typing import Optional
+
     import click
     import click_params
 
     import mdio
-    from typing import Optional
 except SystemError:
     pass
 

diff --git a/tests/test_main.py b/tests/test_main.py
@@ -31,7 +31,7 @@ def test_main_succeeds(runner: CliRunner, segy_input: str, zarr_tmp: Path) -> No
 @pytest.mark.dependency(depends=["test_main_succeeds"])
 def test_main_info_succeeds(runner: CliRunner, zarr_tmp: Path) -> None:
     """It exits with a status code of zero."""
-    cli_args = ["utility", "info"]
+    cli_args = ["info"]
     cli_args.extend(["-i", str(zarr_tmp)])
 
     result = runner.invoke(__main__.main, args=cli_args)
@@ -41,7 +41,7 @@ def test_main_info_succeeds(runner: CliRunner, zarr_tmp: Path) -> None:
 @pytest.mark.dependency(depends=["test_main_succeeds"])
 def test_main_copy_succeeds(runner: CliRunner, zarr_tmp: Path, zarr_tmp2: Path) -> None:
     """It exits with a status code of zero."""
-    cli_args = ["utility", "copy"]
+    cli_args = ["copy"]
     cli_args.extend(["-i", str(zarr_tmp)])
     cli_args.extend(["-o", str(zarr_tmp2)])