import io

from conductor.types.model import Model


class MockModel(Model):
    def __init__(self):
        super().__init__()

    def train(self):
        return

    def predict(self, data: io.StringIO) -> io.StringIO:
        return data


def test_model_constructor():
    model = MockModel()
    assert model.train_env is not None
