Add comparison operators for SizeWithConfidence and OffsetWithConfidence

1 files changed, 42 insertions, 0 deletions

diff --git a/python/types.py b/python/types.py
index 2a141846..c5ce1924 100644
--- a/python/types.py
+++ b/python/types.py
@@ -346,6 +346,27 @@ class OffsetWithConfidence:
 	def __int__(self):
 		return self.value
 
+	def __eq__(self, other):
+		if not isinstance(other, self.__class__):
+			return self.value == int(other)
+		else:
+			return (self.value, self.confidence) == (other.value, other.confidence)
+
+	def __ne__(self, other):
+		return not (self == other)
+
+	def __gt__(self, other):
+		return self.value > int(other)
+
+	def __le__(self, other):
+		return self.value <= int(other)
+
+	def __ge__(self, other):
+		return self.value >= int(other)
+
+	def __lt__(self, other):
+		return self.value < int(other)
+
 	def _to_core_struct(self) -> core.BNOffsetWithConfidence:
 		result = core.BNOffsetWithConfidence()
 		result.value = self.value
@@ -419,6 +440,27 @@ class SizeWithConfidence:
 	def __int__(self):
 		return self.value
 
+	def __eq__(self, other):
+		if not isinstance(other, self.__class__):
+			return self.value == int(other)
+		else:
+			return (self.value, self.confidence) == (other.value, other.confidence)
+
+	def __ne__(self, other):
+		return not (self == other)
+
+	def __gt__(self, other):
+		return self.value > int(other)
+
+	def __le__(self, other):
+		return self.value <= int(other)
+
+	def __ge__(self, other):
+		return self.value >= int(other)
+
+	def __lt__(self, other):
+		return self.value < int(other)
+
 	def _to_core_struct(self) -> core.BNSizeWithConfidence:
 		result = core.BNSizeWithConfidence()
 		result.value = self.value