diff --git a/tests/persistence/test_inmemory.py b/tests/persistence/test_inmemory.py index b0db46e7..9e3141de 100644 --- a/tests/persistence/test_inmemory.py +++ b/tests/persistence/test_inmemory.py @@ -1,11 +1,9 @@ # -*- coding: utf-8 -*- import unittest -from ..utils import ProcessWithCheckpoint - import plumpy -import plumpy +from ..utils import ProcessWithCheckpoint class TestInMemoryPersister(unittest.TestCase): diff --git a/tests/persistence/test_pickle.py b/tests/persistence/test_pickle.py index dd68b4fd..da4ede51 100644 --- a/tests/persistence/test_pickle.py +++ b/tests/persistence/test_pickle.py @@ -5,10 +5,10 @@ if getattr(tempfile, 'TemporaryDirectory', None) is None: from backports import tempfile -from ..utils import ProcessWithCheckpoint - import plumpy +from ..utils import ProcessWithCheckpoint + class TestPicklePersister(unittest.TestCase): def test_save_load_roundtrip(self): diff --git a/tests/rmq/test_process_comms.py b/tests/rmq/test_process_comms.py index 4c7a4f1a..c6826a24 100644 --- a/tests/rmq/test_process_comms.py +++ b/tests/rmq/test_process_comms.py @@ -1,6 +1,5 @@ # -*- coding: utf-8 -*- import asyncio -import copy import kiwipy import pytest diff --git a/tests/test_communications.py b/tests/test_communications.py index 37177d6e..f7e04255 100644 --- a/tests/test_communications.py +++ b/tests/test_communications.py @@ -4,8 +4,6 @@ import pytest from kiwipy import CommunicatorHelper -import pytest -from kiwipy import CommunicatorHelper from plumpy.communications import LoopCommunicator diff --git a/tests/test_expose.py b/tests/test_expose.py index f48ce32e..c5e6014c 100644 --- a/tests/test_expose.py +++ b/tests/test_expose.py @@ -1,47 +1,11 @@ # -*- coding: utf-8 -*- import unittest -from .utils import NewLoopProcess - from plumpy.ports import PortNamespace from plumpy.process_spec import ProcessSpec from plumpy.processes import Process - -def validator_function(input, port): - pass - - -class BaseNamespaceProcess(NewLoopProcess): - @classmethod - def define(cls, spec): - super().define(spec) - spec.input('top') - spec.input('namespace.sub_one') - spec.input('namespace.sub_two') - spec.inputs['namespace'].valid_type = (int, float) - spec.inputs['namespace'].validator = validator_function - - -class BaseProcess(NewLoopProcess): - @classmethod - def define(cls, spec): - super().define(spec) - spec.input('a', valid_type=str, default='a') - spec.input('b', valid_type=str, default='b') - spec.inputs.dynamic = True - spec.inputs.valid_type = str - - -class ExposeProcess(NewLoopProcess): - @classmethod - def define(cls, spec): - super().define(spec) - spec.expose_inputs(BaseProcess, namespace='base.name.space') - spec.input('c', valid_type=int, default=1) - spec.input('d', valid_type=int, default=2) - spec.inputs.dynamic = True - spec.inputs.valid_type = int +from .utils import NewLoopProcess def validator_function(input, port): diff --git a/tests/test_process_comms.py b/tests/test_process_comms.py index c59737ac..44947230 100644 --- a/tests/test_process_comms.py +++ b/tests/test_process_comms.py @@ -1,9 +1,9 @@ # -*- coding: utf-8 -*- import pytest -from tests import utils import plumpy from plumpy import process_comms +from tests import utils class Process(plumpy.Process): diff --git a/tests/test_processes.py b/tests/test_processes.py index cec20c51..eb5bf599 100644 --- a/tests/test_processes.py +++ b/tests/test_processes.py @@ -3,19 +3,16 @@ import asyncio import enum -from plumpy.process_comms import KillMessage import unittest import kiwipy import pytest -from tests import utils - -import plumpy -import pytest import plumpy from plumpy import BundleKeys, Process, ProcessState +from plumpy.process_comms import KillMessage from plumpy.utils import AttributesFrozendict +from tests import utils class ForgetToCallParent(plumpy.Process):