1515from imdclient .IMDClient import imdframe_memsize , IMDClient
1616from imdclient .IMDProtocol import IMDHeaderType
1717from .utils import (
18- get_free_port ,
1918 create_default_imdsinfo_v3 ,
2019)
2120from .server import InThreadIMDServer
4645
4746class TestIMDClientV3 :
4847
49- @pytest .fixture
50- def port (self ):
51- return get_free_port ()
52-
5348 @pytest .fixture
5449 def universe (self ):
5550 return mda .Universe (COORDINATES_TOPOLOGY , COORDINATES_H5MD )
@@ -59,13 +54,13 @@ def imdsinfo(self):
5954 return create_default_imdsinfo_v3 ()
6055
6156 @pytest .fixture
62- def server_client_two_frame_buf (self , universe , imdsinfo , port ):
57+ def server_client_two_frame_buf (self , universe , imdsinfo ):
6358 server = InThreadIMDServer (universe .trajectory )
6459 server .set_imdsessioninfo (imdsinfo )
65- server .handshake_sequence ("localhost" , port , first_frame = False )
60+ server .handshake_sequence ("localhost" , first_frame = False )
6661 client = IMDClient (
6762 f"localhost" ,
68- port ,
63+ server . port ,
6964 universe .trajectory .n_atoms ,
7065 buffer_size = imdframe_memsize (universe .trajectory .n_atoms , imdsinfo )
7166 * 2 ,
@@ -76,14 +71,14 @@ def server_client_two_frame_buf(self, universe, imdsinfo, port):
7671 server .cleanup ()
7772
7873 @pytest .fixture (params = [">" , "<" ])
79- def server_client (self , universe , imdsinfo , port , request ):
74+ def server_client (self , universe , imdsinfo , request ):
8075 server = InThreadIMDServer (universe .trajectory )
8176 imdsinfo .endianness = request .param
8277 server .set_imdsessioninfo (imdsinfo )
83- server .handshake_sequence ("localhost" , port , first_frame = False )
78+ server .handshake_sequence ("localhost" , first_frame = False )
8479 client = IMDClient (
8580 f"localhost" ,
86- port ,
81+ server . port ,
8782 universe .trajectory .n_atoms ,
8883 )
8984 server .join_accept_thread ()
@@ -168,13 +163,13 @@ def test_pause_resume_no_disconnect(self, server_client_two_frame_buf):
168163 server .expect_packet (IMDHeaderType .IMD_DISCONNECT )
169164
170165 @pytest .mark .parametrize ("cont" , [True , False ])
171- def test_continue_after_disconnect (self , universe , imdsinfo , port , cont ):
166+ def test_continue_after_disconnect (self , universe , imdsinfo , cont ):
172167 server = InThreadIMDServer (universe .trajectory )
173168 server .set_imdsessioninfo (imdsinfo )
174- server .handshake_sequence ("localhost" , port , first_frame = False )
169+ server .handshake_sequence ("localhost" , first_frame = False )
175170 client = IMDClient (
176171 f"localhost" ,
177- port ,
172+ server . port ,
178173 universe .trajectory .n_atoms ,
179174 continue_after_disconnect = cont ,
180175 )
@@ -198,10 +193,6 @@ def test_incorrect_atom_count(self, server_client_incorrect_atoms, universe):
198193
199194
200195class TestIMDClientV3ContextManager :
201- @pytest .fixture
202- def port (self ):
203- return get_free_port ()
204-
205196 @pytest .fixture
206197 def universe (self ):
207198 return mda .Universe (COORDINATES_TOPOLOGY , COORDINATES_H5MD )
@@ -211,19 +202,19 @@ def imdsinfo(self):
211202 return create_default_imdsinfo_v3 ()
212203
213204 @pytest .fixture
214- def server (self , universe , imdsinfo , port ):
205+ def server (self , universe , imdsinfo ):
215206 server = InThreadIMDServer (universe .trajectory )
216207 server .set_imdsessioninfo (imdsinfo )
217208 yield server
218209 server .cleanup ()
219210
220- def test_context_manager_traj_unchanged (self , server , port , universe ):
221- server .handshake_sequence ("localhost" , port , first_frame = False )
211+ def test_context_manager_traj_unchanged (self , server , universe ):
212+ server .handshake_sequence ("localhost" , first_frame = False )
222213
223214 i = 0
224215 with IMDClient (
225216 "localhost" ,
226- port ,
217+ server . port ,
227218 universe .trajectory .n_atoms ,
228219 ) as client :
229220 server .send_frames (0 , 5 )
0 commit comments