--- title: Title keywords: fastai sidebar: home_sidebar nb_path: "nbs/test_schema.ipynb" ---
{% raw %}
{% endraw %} {% raw %}
%load_ext autoreload
%autoreload 2
{% endraw %} {% raw %}
import inspect
from pymemri.data import _central_schema
from pymemri.pod.client import PodClient
{% endraw %} {% raw %}
central_schema = {
    k: v
    for k, v in inspect.getmembers(_central_schema)
    if inspect.isclass(v) and v.__module__ == "pymemri.data._central_schema"
}
print("num schema classes:", len(central_schema))
num schema classes: 44
{% endraw %} {% raw %}
client = PodClient()

for k, v in central_schema.items():
    assert client.add_to_schema(v), f"Could not add {k} to schema"
{% endraw %} {% raw %}
from pymemri.data.schema import EmailMessage, Country, Receipt, Account
from datetime import datetime

item_1 = EmailMessage(starred=True, content="This is a test", dateSent=datetime.now())
item_2 = Country(name="Belgium")
item_3 = Receipt(category="clothing", store="clothing store", totalCost=100000)
item_4 = Account(displayName="1234", identifier="5678", isMe=False, service="whatsapp")
assert client.bulk_action(create_items=[item_1, item_2, item_3, item_4])
BULK: Writing 4/4 items/edges
Completed Bulk action, written 4 items/edges
{% endraw %}