Add tests for sphinx extension