"""Tests for header elements."""

from aki_prj23_transparenzregister.ui import header_elements


def test_import() -> None:
    """Checks if an import co ui_elements can be made."""
    assert header_elements is not None


def test_create_header() -> None:
    """Checks if the header can be created."""
    options = {1: "a", 2: "b"}
    header_elements.create_header(options)


def test_create_selection_header() -> None:
    """Checks if the company header can be created."""
    selected_company = "Test GmbH"
    header_elements.create_selection_header(selected_company)