Skip to content

Commit 5dd7081

Browse files
committed
Added Packet().__eq__ method
1 parent f308969 commit 5dd7081

File tree

2 files changed

+11
-7
lines changed

2 files changed

+11
-7
lines changed

tests/test_hlapi.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff 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)

tsip/hlapi.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff 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.

0 commit comments

Comments
 (0)