Improving test coverage for schema.py to 100% coverage