Improve test coverage

[pohlm01]

Codecov provides a nice overview
Any contribution improving the test coverage is welcome, even if it only adds a (small) part.

[joaoag]

Spoke to @stefanvi at the monthly LF catch-up call and he suggested this would be good to pick up while I'm waiting to hear back from #93, so assigning this to myself