File size: 2,014 Bytes
16760fa b1390b1 16760fa b1390b1 16760fa b1390b1 16760fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | from __future__ import annotations
import numpy as np
from toaster.core import LabelClass, LabelSchema
from toaster.persistence import LabelStore, SchemaStore, SessionState, SessionStore
def test_schema_store_roundtrip(tmp_path):
store = SchemaStore()
source = tmp_path / "scan.ply"
schema = LabelSchema(
classes=[LabelClass(0, "unlabeled", (0, 0, 0)), LabelClass(1, "tree", (1, 2, 3))],
unlabeled_id=0,
)
out = store.save(source, schema, cloud_path=source)
assert out == store.path_for(source)
assert out.name == "scan_toaster_schema.yaml" # extension dropped, '_'-separated
assert "cloud:" in out.read_text() # records the originating cloud path
loaded = store.load(source)
assert loaded is not None
assert loaded.get(1).name == "tree"
assert loaded.get(1).color == (1, 2, 3)
def test_schema_store_missing_returns_none(tmp_path):
assert SchemaStore().load(tmp_path / "absent.ply") is None
def test_label_store_roundtrip(tmp_path):
store = LabelStore()
source = tmp_path / "scan.ply"
labels = np.array([0, 1, 2, 1], dtype=np.int32)
out = store.save(source, labels)
assert out == store.path_for(source)
assert out.name == "scan_toaster.npy" # extension dropped, '_'-separated
loaded = store.load(source)
assert np.array_equal(loaded, labels)
def test_label_store_missing_returns_none(tmp_path):
assert LabelStore().load(tmp_path / "absent.ply") is None
def test_session_store_roundtrip(tmp_path):
store = SessionStore(tmp_path / "session.json")
state = SessionState(cloud_path="a.ply", active_class=3)
state.remember("a.ply")
state.remember("b.ply")
store.save(state)
loaded = store.load()
assert loaded.active_class == 3
assert loaded.recent_files[:2] == ["b.ply", "a.ply"]
def test_session_store_missing_returns_defaults(tmp_path):
state = SessionStore(tmp_path / "none.json").load()
assert state.active_class == 0
assert state.recent_files == []
|