Write unit tests using Python unit testing tools

Currently the unit tests are written using a shell script, but they should use Python's unit testing tools. This will make it easier to integrate into other tools such as continuous integration, performance testing, etc.