--- title: Title keywords: fastai sidebar: home_sidebar nb_path: "nbs/test_schema.ipynb" ---
%load_ext autoreload
%autoreload 2
import inspect
from pymemri.data import _central_schema
from pymemri.pod.client import PodClient
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))
client = PodClient()
for k, v in central_schema.items():
assert client.add_to_schema(v), f"Could not add {k} to schema"
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])