Practicing python unit tests, feel free to contribute/use as an example