tests/src/test_utils.py

from src.display.utils import (
    COLS_LONG_DOC,
    COLS_QA,
    TYPES_LONG_DOC,
    TYPES_QA,
    AutoEvalColumnQA,
    fields,
    get_default_auto_eval_column_dict,
)


def test_fields():
    for c in fields(AutoEvalColumnQA):
        print(c)


def test_macro_variables():
    print(f"COLS_QA: {COLS_QA}")
    print(f"COLS_LONG_DOC: {COLS_LONG_DOC}")
    print(f"TYPES_QA: {TYPES_QA}")
    print(f"TYPES_LONG_DOC: {TYPES_LONG_DOC}")


def test_get_default_auto_eval_column_dict():
    auto_eval_column_dict_list = get_default_auto_eval_column_dict()
    assert len(auto_eval_column_dict_list) == 9