1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
|
import pytest
from jupyter_cache.cache.db import NbCacheRecord, Setting, create_db
def test_setting(tmp_path):
db = create_db(tmp_path)
Setting.set_value("a", 1, db)
assert Setting.get_value("a", db) == 1
assert Setting.get_dict(db) == {"a": 1}
def test_nb_record(tmp_path):
db = create_db(tmp_path)
bundle = NbCacheRecord.create_record("a", "b", db)
assert bundle.hashkey == "b"
with pytest.raises(ValueError):
NbCacheRecord.create_record("a", "b", db)
NbCacheRecord.create_record("a", "c", db, data="a")
assert NbCacheRecord.record_from_hashkey("b", db).uri == "a"
assert {b.hashkey for b in NbCacheRecord.records_from_uri("a", db)} == {"b", "c"}
|