Added bandit based unit tests
Improved test harness