How to mock object for testing -- update tests

[ba-talibe]

Update tests with a way to mock tests. See python unittest module: https://docs.python.org/3/library/unittest.html