Improve tests coverage

[Pooya Parsa]

Some nice initiatives from #86 by @alex-key which could be inspired.

[Alex Korytskyi]

Hi @pi0 , I will take a look and create a new PR to keep tests inprovements