Raise coverage percent up to 90

[ApusBerliozi]

Right now our test coverage is pretty low, 68%. We need to cover more, at least 90.

It would be better to split this task, and do it iteratively via multiple PR's

[ParallelVoid]

I'm taking a look at this right now.

[ApusBerliozi]

@ParallelVoid oi, thx mate!

Assigned the issue to ya

[ParallelVoid]

First PR for this, tackled the agents folder first: #158

[ApusBerliozi]

@ParallelVoid, hello!

Could you please provide us with an ETA for this task? Just for us to know

[ParallelVoid]

I'm working on the PR for nn module, after which I believe the code coverage will go up to 90%. So I would guess end of day, unless codecov has weird calculations (which it must have, as it does not align with pytest code coverage calculations).

[ParallelVoid]

Second PR: #204

[ParallelVoid]

[ParallelVoid]

do we need a third pr?