Integration tests needed for each module and main

[kchawla-pi]

Usually use standard python, will consider using a testing framework like pytest