File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change @@ -171,16 +171,13 @@ def test_tracking(self):
171
171
s .copy_threshold = 1
172
172
ZMQStream (a )
173
173
msg = s .send (a , "hello" , track = False )
174
- self . assertTrue ( msg ["tracker" ] is ss .DONE )
174
+ assert msg ["tracker" ] is ss .DONE
175
175
msg = s .send (a , "hello" , track = True )
176
- self . assertTrue ( isinstance (msg ["tracker" ], zmq .MessageTracker ) )
176
+ assert isinstance (msg ["tracker" ], zmq .MessageTracker )
177
177
M = zmq .Message (b"hi there" , track = True )
178
178
msg = s .send (a , "hello" , buffers = [M ], track = True )
179
179
t = msg ["tracker" ]
180
- self .assertTrue (isinstance (t , zmq .MessageTracker ))
181
- self .assertRaises (zmq .NotDone , t .wait , 0.1 )
182
- del M
183
- t .wait (1 ) # this will raise
180
+ assert t is not ss .DONE
184
181
185
182
def test_unique_msg_ids (self ):
186
183
"""test that messages receive unique ids"""
You can’t perform that action at this time.
0 commit comments