rewrite: implement CALL_METHOD.

rchen152 · rchen152 · commit 02a4afd79fb9 · 2024-03-14T15:03:49.000-07:00
diff --git a/pytype/rewrite/frame.py b/pytype/rewrite/frame.py
@@ -430,6 +430,13 @@ def byte_CALL_FUNCTION(self, opcode):
     func_var = self._stack.pop()
     self._call_function(func_var, abstract.Args(posargs=args))
 
+  def byte_CALL_METHOD(self, opcode):
+    args = self._stack.popn(opcode.arg)
+    func_var = self._stack.pop()
+    # pop the NULL off the stack (see LOAD_METHOD)
+    self._stack.pop_and_discard()
+    self._call_function(func_var, abstract.Args(posargs=args))
+
   def byte_POP_TOP(self, opcode):
     del opcode  # unused
     self._stack.pop_and_discard()
