Create unit tests with unittest

[matta174]

https://docs.python.org/3/library/unittest.html