import pytest

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


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


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