diff --git a/ipyparallel/shellcmd.py b/ipyparallel/shellcmd.py index 3f76c725..5755b2a4 100644 --- a/ipyparallel/shellcmd.py +++ b/ipyparallel/shellcmd.py @@ -16,7 +16,7 @@ import sys from argparse import ArgumentParser -from .cluster.shellcmd import ShellCommandReceive +from .cluster.shellcmd_receive import ShellCommandReceive def main(): diff --git a/ipyparallel/tests/test_shellcmd.py b/ipyparallel/tests/test_shellcmd.py index 748dd83c..6fce5d51 100644 --- a/ipyparallel/tests/test_shellcmd.py +++ b/ipyparallel/tests/test_shellcmd.py @@ -128,6 +128,12 @@ ] +def test_shellcmd_module_import(): + import ipyparallel.shellcmd + + assert callable(ipyparallel.shellcmd.main) + + @pytest.fixture def shellcmd_test_cmd(): """returns a command that runs for 5 seconds"""