Add Unit Testing

Using PyTest, write unit tests. Include code coverage.