Write unit tests

[artitw]

Follow guidelines from official Python documentation for unit testing: https://docs.python.org/3/library/unittest.html