Is your feature request related to a problem? Please describe.
The basic system-level test suite added in PR #38 only has a few basic test cases.
Describe the solution you'd like
Gather requirements / coordinate a more comprehensive collection of tests for the various tasks (dataset creation, training, etc.)
Describe alternatives you've considered
No response
Additional context
No response
Organisation
No response