Memory leak in jsonschema

jsonschema_memory_leak.py

import jsonschema

import os
import psutil

import matplotlib.pyplot as plt

schema = {
    "$id": "https://example.com/person.schema.json",
    "$schema": "https://json-schema.org/draft/2020-12/schema",
    "title": "Person",
    "type": "object",
    "properties": {
        "firstName": {
            "type": "string",
            "description": "The person's first name."
        },
        "lastName": {
            "type": "string",
            "description": "The person's last name."
        },
        "age": {
            "description": "Age in years which must be equal to or greater than zero.",
            "type": "integer",
            "minimum": 0
        }
    }
}


instance = {
    "firstName": "John",
    "lastName": "Smith",
    "age": 40
}


memory_usage = []
process = psutil.Process(os.getpid())
iterations = range(0, 1000)

for i in iterations:
    # Call validate from JSON schema.
    jsonschema.validate(instance, schema)

    # Track the memory usage in Megabytes.
    memory = process.memory_info()
    memory_usage.append(memory.rss / 1e6)


fig, ax = plt.subplots()
ax.plot(iterations, memory_usage)
ax.set_title("Possible memory leak in jsonschema")
ax.set_ylabel("Memory Usage (Mb)")
ax.set_xlabel("Number of times `validate()` is called.")
fig.savefig("test.png")