Added __eq__ to MediumLevelILOperationAndSize as well

1 files changed, 6 insertions, 0 deletions

diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index 103d8c5f..f35b2e6e 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -71,6 +71,12 @@ class MediumLevelILOperationAndSize(object):
 			return "<%s>" % self.operation.name
 		return "<%s %d>" % (self.operation.name, self.size)
 
+	def __eq__(self, other):
+		if isinstance(other, MediumLevelILOperation):
+			return other == self.operation
+		if isinstance(other, MediumLevelILOperationAndSize):
+			return other.size == self.size and other.operation == self.operation
+
 
 class MediumLevelILInstruction(object):
 	"""