3030
3131from common import dask_env , ray_env
3232
33- from deisa .common import DeisaInterface , BridgeInterface
33+ from deisa .common import IDeisa , IBridge
3434
3535
3636@pytest .mark .parametrize ("env_setup" , ["dask_env" , "ray_env" ])
@@ -39,7 +39,7 @@ def test_deisa_api(request, env_setup):
3939 from deisa .dask import Deisa
4040
4141 client , cluster = request .getfixturevalue (env_setup )
42- deisa : DeisaInterface = Deisa (get_connection_info = lambda : client , wait_for_go = False )
42+ deisa : IDeisa = Deisa (get_connection_info = lambda : client , wait_for_go = False )
4343
4444 assert hasattr (deisa , 'close' ) and callable (deisa .close )
4545 assert hasattr (deisa , 'get_array' ) and callable (deisa .get_array )
@@ -55,9 +55,9 @@ def test_bridge_api(request, env_setup):
5555 from deisa .dask import Bridge
5656
5757 client , cluster = request .getfixturevalue (env_setup )
58- bridge : BridgeInterface = Bridge (id = 0 ,
59- arrays_metadata = {},
60- system_metadata = {'connection' : client , 'nb_bridges' : 1 },
61- wait_for_go = False )
58+ bridge : IBridge = Bridge (id = 0 ,
59+ arrays_metadata = {},
60+ system_metadata = {'connection' : client , 'nb_bridges' : 1 },
61+ wait_for_go = False )
6262
6363 assert hasattr (bridge , 'send' ) and callable (bridge .send )
0 commit comments