The example in [re_generative.py](https://github.com/ChristophAlt/pytorch-ie/blob/main/examples/predict/re_generative.py) may actually help to write the tests.