test_project_set_model_setup_complete.py

import pytest
from lbox.exceptions import LabelboxError, OperationNotAllowedException


def test_live_chat_evaluation_project(
    live_chat_evaluation_project, model_config
):
    project = live_chat_evaluation_project

    project.set_project_model_setup_complete()
    assert bool(project.model_setup_complete) is True

    with pytest.raises(
        expected_exception=LabelboxError,
        match="Cannot create model config for project because model setup is complete",
    ):
        project.add_model_config(model_config.uid)


def test_live_chat_evaluation_project_delete_cofig(
    live_chat_evaluation_project, model_config
):
    project = live_chat_evaluation_project
    project_model_config_id = project.add_model_config(model_config.uid)
    assert project_model_config_id

    project_model_config = None
    for pmg in project.project_model_configs():
        if pmg.uid == project_model_config_id:
            project_model_config = pmg
            break
    assert project_model_config

    project.set_project_model_setup_complete()
    assert bool(project.model_setup_complete) is True

    with pytest.raises(
        expected_exception=LabelboxError,
        match="Cannot (create model config for project because model setup is complete|perform this action because model setup is complete)",
    ):
        project_model_config.delete()


def test_offline_chat_evaluation_project(
    offline_chat_evaluation_project, model_config
):
    project = offline_chat_evaluation_project

    with pytest.raises(
        expected_exception=OperationNotAllowedException,
        match="Only live model chat evaluation projects can complete model setup",
    ):
        project.set_project_model_setup_complete()


def test_any_other_project(project, model_config):
    with pytest.raises(
        expected_exception=OperationNotAllowedException,
        match="Only live model chat evaluation projects can complete model setup",
    ):
        project.set_project_model_setup_complete()