from bson import encode
from bson.errors import InvalidDocument

def main():
    small_str = "x" * 1024
    while True:
        for _ in range (10000):
            doc = {"data": small_str, "invalid": set()}
            try:
                encode(doc)
                raise Exception("Shouldn't get here")
            except InvalidDocument as exc:
                pass
                # assert exc.document == doc

if __name__ == "__main__":
    main()