PyAuto test for GPU memory consumption

chrome/test/pyautolib/pyauto.py

Index: chrome/test/pyautolib/pyauto.py
diff --git a/chrome/test/pyautolib/pyauto.py b/chrome/test/pyautolib/pyauto.py
index 53d86ca6deccec640e6509ae856df418e1016e8d..0790bcf9dabf7e3de8e84c1a3a0dcb4424cf7844 100755
--- a/chrome/test/pyautolib/pyauto.py
+++ b/chrome/test/pyautolib/pyauto.py
@@ -3830,6 +3830,26 @@ class PyUITest(pyautolib.PyUITestBase, unittest.TestCase):
     }
     return self._GetResultFromJSONRequest(cmd_dict, windex=windex)
 
+  def GetGpuMemStats(self, windex=0):
+    """Returns the current FPS associated with the renderer process for a tab.
+
+    FPS is the rendered frames per second.
+
+    Args:
+      window_index: The window index, default is 0.
+
+    Returns:
+      A dictionary containing GPU memory usage info.
+      Example:
+        { 'gpu_active': true,
+          'gpu_memory_bytes_total': 1073741824,
+          'gpu_memory_bytes_used': 16777216 }
+    """
+    cmd_dict = {  # Prepare command for the json interface.
+      'command': 'GetGpuMemStats',
+    }
+    return self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+
   def IsFullscreenForBrowser(self, windex=0):
     """Returns true if the window is currently fullscreen and was initially
     transitioned to fullscreen by a browser (vs tab) mode transition."""