Code coverage

[t-kimber]

Update tests, ideally have 80% coverage.

Note: have a check for estimate input for forward method in pytorch models, see #53