Unit tests needed for each function.

[kchawla-pi]

Have to write unit tests for all the individual wrapper, runner and helper functions across all modules.
Have used standard python so far. Will look into using pytests or something else.