Added pydoc back to unit test file