  def _StoreDiskForCleanup(self, disk):
    """Store a disk for cleanup when tests finish.

    Args:
      disk (AZDisk): An Azure disk.
    """
    self.disks.append(disk)
Store a disk for cleanup when tests finish.

Args:
disk (AZDisk): An Azure disk.
tests/providers/azure/e2e_cli.py