Поиск первого элемента из списка с помощью python

Вопрос следующий: Пишу автотест. Мне нужно зайти на яндекс.картинки и
1) найти Первую картинку из имеющихся, 2) открыть её, 3) перейти назад, 4) опять найти первую 5) убедиться, что это одна и та же картинка (из шага 2)

Пока нашел:

    from time import sleep
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.remote.webelement import WebElement
driver = webdriver.Chrome("C:\\Users\\Admin\\Documents\\trash\\Python projects\\chromedriver.exe")
sleep(2)
driver.maximize_window()
driver.get("https://yandex.ru")
sleep(2)
a: WebElement=driver.find_element_by_link_text("Картинки")
assert a, ("ОПА")##Проверка что есть картинки
a.send_keys(Keys.ENTER)
sleep(2)
b: WebElement=driver.find_element_by_class_name("cl-teaser[1]")          
b.click()
sleep(2)
driver.find_element_by_class_name("layout__nav__left")
c: WebElement=driver.find_element_by_css_selector("div.masonry__column:nth-child(1) > div:nth-child(1").click()
assert b