import pytest

from image_prediction.formatter.formatters.camel_case import Snake2CamelCaseKeyFormatter
from image_prediction.formatter.formatters.enum import EnumFormatter


def test_enum_formatter(metadata, metadata_formatted):
    assert list(EnumFormatter()(metadata)) == metadata_formatted


@pytest.mark.parametrize("label_format", ["probability"])
def test_enum_formatter(metadata_plus_mapped_prediction, metadata_formatted_plus_mapped_prediction_formatted):
    assert list(EnumFormatter()(metadata_plus_mapped_prediction)) == metadata_formatted_plus_mapped_prediction_formatted


def test_camel_case_key_formatter(snake_case_data, camel_case_data):
    assert Snake2CamelCaseKeyFormatter()(snake_case_data) == camel_case_data


@pytest.fixture
def snake_case_data():
    return {"a_key": {"key": None, "key_2": ["may_not_be_changed", (1, 2, 2.2)]}, 2: {"yet_another_key": 3, 4: "a"}}


@pytest.fixture
def camel_case_data():
    return {"aKey": {"key": None, "key2": ["may_not_be_changed", (1, 2, 2.2)]}, 2: {"yetAnotherKey": 3, 4: "a"}}