import pytest

from image_prediction.formatter.formatters.enum import EnumFormatter
from image_prediction.formatter.formatters.identity import IdentityFormatter
from image_prediction.compositor.compositor import FormatterCompositor


def test_single_formatter(metadata):
    compositor = FormatterCompositor(IdentityFormatter())
    assert metadata == compositor(metadata)


def test_two_formatters(metadata, metadata_formatted):
    compositor = FormatterCompositor(IdentityFormatter(), EnumFormatter())
    assert metadata_formatted == list(compositor(metadata))