how to find screen size in pygame code example

Example: how to get pygame window height size

import pygame
pygame.init()

window = pygame.display.set_mode((500, 500))
x, y = window.get_size()

print("The width is %s" %x)
print("The height is %s" %y)