Increase the test coverage in adding more tests

[pythonbrad]

Problem

Currently the coverage is very low.

Solution

• Add more unit-tests
• Add one integration tests (test for the cli)