23 lines
714 B
Python
23 lines
714 B
Python
import pytest
|
|
|
|
from image_prediction.formatter.formatters.camel_case import Snake2CamelCaseKeyFormatter
|
|
from image_prediction.formatter.formatters.enum import EnumFormatter
|
|
|
|
|
|
def test_formatter(metadata, metadata_formatted):
|
|
assert list(EnumFormatter()(metadata)) == metadata_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"]}, 2: {"yet_another_key": None}}
|
|
|
|
|
|
@pytest.fixture
|
|
def camel_case_data():
|
|
return {"aKey": {"key": None, "key2": ["may_not_be_changed"]}, 2: {"yetAnotherKey": None}}
|