File tree Expand file tree Collapse file tree 2 files changed +11
-7
lines changed
Expand file tree Collapse file tree 2 files changed +11
-7
lines changed Original file line number Diff line number Diff line change @@ -27,13 +27,12 @@ def setup(self):
2727 self .pkt6 = Packet .unpack (self .pkt3 .pack ())
2828
2929 def test_pack (self ):
30- assert self .pkt1 .fields == \
31- self .fields == \
32- self .pkt2 .fields == \
33- self .pkt3 .fields == \
34- self .pkt4 .fields == \
35- self .pkt5 .fields == \
36- self .pkt6 .fields
30+ assert self .pkt1 == \
31+ self .pkt2 == \
32+ self .pkt3 == \
33+ self .pkt4 == \
34+ self .pkt5 == \
35+ self .pkt6
3736
3837
3938@raises (PackError )
Original file line number Diff line number Diff line change @@ -59,6 +59,11 @@ def __init__(self, *fields):
5959 except TypeError :
6060 self .fields = fields
6161
62+ # Packets are equal if their fields are equal
63+ #
64+ def __eq__ (self , other ):
65+ return self .fields == other .fields
66+
6267
6368 # Make self.fields accessible as indexes on the
6469 # packet instance.
You can’t perform that action at this time.
0 commit comments