TST: Create integration tests for example use cases

[schobbejak]

Create a list of example use cases and write integration tests. This would give a better indicator of whether the library still works as desired
Examples:

• Train a torch model using the torch trainer
• Create a transformation block that multiplies an array by 2