def METHOD_NAME(self, pytorch_model, series):
    model = pytorch_model(
        input_chunk_length=1, output_chunk_length=3, n_epochs=1, **tfm_kwargs
    )
    model.fit(series)
    pred = model.predict(7)
    assert len(pred) == 7
    pred = model.predict(2)
    assert len(pred) == 2
    assert pred.width == 1
    pred = model.predict(4)
    assert len(pred) == 4
    assert pred.width == 1