bpo-39902: support equality comparisons in dis.Bytecode

isidentical · isidentical · commit c1bf01bb63cc · 2020-05-17T05:06:18.000+03:00
diff --git a/Doc/whatsnew/3.9.rst b/Doc/whatsnew/3.9.rst
@@ -325,6 +325,12 @@ and :meth:`~datetime.datetime.isocalendar()` of :class:`datetime.datetime`
 methods now returns a :func:`~collections.namedtuple` instead of a :class:`tuple`.
 (Contributed by Dong-hee Na in :issue:`24416`.)
 
+dis
+---
+
+:class:`dis.Bytecode` objects now support equality comparisons. (Contributed
+by Batuhan Taskaya in :issue:`39902`.)
+
 fcntl
 -----
 
diff --git a/Lib/dis.py b/Lib/dis.py
@@ -536,6 +536,16 @@ def dis(self):
                                lasti=offset)
             return output.getvalue()
 
+    def __eq__(self, other):
+        if isinstance(other, Bytecode):
+            return (self.codeobj, self.first_line, self.current_offset) == (
+                other.codeobj,
+                other.first_line,
+                other.current_offset,
+            )
+        else:
+            return NotImplemented
+
 
 def _test():
     """Simple test program to disassemble a file."""
diff --git a/Lib/test/test_dis.py b/Lib/test/test_dis.py
@@ -1202,5 +1202,16 @@ def test_from_traceback_dis(self):
         b = dis.Bytecode.from_traceback(tb)
         self.assertEqual(b.dis(), dis_traceback)
 
+    def test_equality(self):
+        self.assertEqual(dis.Bytecode("print(1)"), dis.Bytecode("print(1)"))
+        self.assertNotEqual(dis.Bytecode("print(1)"), dis.Bytecode("print(2)"))
+        self.assertNotEqual(
+            dis.Bytecode("print(1)", first_line=3), dis.Bytecode("print(1)")
+        )
+        self.assertNotEqual(
+            dis.Bytecode("print(1)", current_offset=5),
+            dis.Bytecode("print(1)"),
+        )
+
 if __name__ == "__main__":
     unittest.main()
diff --git a/Misc/NEWS.d/next/Library/2020-03-08-15-16-45.bpo-39902.HoE484.rst b/Misc/NEWS.d/next/Library/2020-03-08-15-16-45.bpo-39902.HoE484.rst
@@ -0,0 +1,2 @@
+Added support for equality comparisons in ``dis.Bytecode`` objects. Patch by
+Batuhan Taskaya.

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	+Added support for equality comparisons in ``dis.Bytecode`` objects. Patch by
	`2`	`+Batuhan Taskaya.`