Автоматизированное тестирование selenium

from selenium.webdriver.common.by import By
from selenium import webdriver
import time
import math
import pytest
 
@pytest.fixture(scope="class")
def browser():
    browser = webdriver.Chrome()
    browser.implicitly_wait(6)
    yield browser
    browser.quit()
    return browser
 
@pytest.mark.parametrize('number', ["236895", "236896", "236897", "236898", "236899", "236903", "236904", "236905"])
class TestMain():
    def test_case(self, number, browser):
        link = f"https://68568/lesson/{number}/step/1"
        browser.get(link)
        answer = str(math.log(int(time.time())))
        browser.find_element(By.CSS_SELECTOR, 'textarea').send_keys(answer)
        browser.find_element(By.CSS_SELECTOR, "button.submit-submission").click()
        message = browser.find_element(By.CSS_SELECTOR, "pre.smart-hints__hint").text
        if message != "Correct!":
            print(message)
        assert "Correct!" == message

def procedure(x: int, y: int):
    print(x + y)
 
 
def function(x: int, y: int) -> int:
    return x + y
 
 
class MY_CLASS:
 
    def __init__(self, x: int, y: int):
        self.x, self.y = x, y
 
    def method(self) -> int:
        return self.x + self.y
 
 
procedure(10, 20)
 
print(function(30, 40))
 
my_class = MY_CLASS(50, 60)
print(my_class.method())