"""Tests for company elements."""

from sqlalchemy.orm import Session

from aki_prj23_transparenzregister.ui import company_elements, data_elements


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


def test_create_company_stats(full_db: Session) -> None:
    """Checks if the company widgets can be created."""
    company_df = data_elements.get_company_data(full_db)
    value_chosen = 1
    selected_company_stats = company_df.loc[value_chosen]
    company_elements.create_company_stats(selected_company_stats)


def test_create_tabs(full_db: Session) -> None:
    """Checks if the tabs of the company page can be created."""
    selected_company_id = 1
    company_elements.create_tabs(full_db, selected_company_id)