16 lines
571 B
Python
16 lines
571 B
Python
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))
|