Improve unit test coverage

[JulesBelveze]

We already have a decent amount of unit tests but they are only set for a specific configuration.

Example: we have a lot of tests for NERSample but none for QASample, SummarizationSample, ...