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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
"""Unit tests for annotation import/export."""
import tempfile
from pathlib import Path
from sigima.io import read_annotations, write_annotations
from sigima.objects.signal.creation import create_signal
def test_write_read_annotations():
"""Test writing and reading annotations to/from file."""
# Create test annotations
annotations = [
{
"type": "plotpy_item",
"item_class": "AnnotatedRectangle",
"plotpy_json": "{}",
},
{
"type": "plotpy_item",
"item_class": "AnnotatedCircle",
"plotpy_json": "{}",
},
]
# Write to temporary file
with tempfile.NamedTemporaryFile(suffix=".dlabann", delete=False) as f:
filepath = f.name
try:
write_annotations(filepath, annotations)
# Read back
read_ann = read_annotations(filepath)
# Verify
assert read_ann == annotations
assert len(read_ann) == 2
assert read_ann[0]["item_class"] == "AnnotatedRectangle"
assert read_ann[1]["item_class"] == "AnnotatedCircle"
finally:
# Clean up
Path(filepath).unlink(missing_ok=True)
def test_write_read_annotations_with_object():
"""Test writing and reading annotations using signal object."""
# Create signal with annotations
obj = create_signal("Test")
obj.set_annotations(
[
{"type": "label", "text": "Peak 1"},
{"type": "label", "text": "Peak 2"},
]
)
# Write to temporary file
with tempfile.NamedTemporaryFile(suffix=".dlabann", delete=False) as f:
filepath = f.name
try:
write_annotations(filepath, obj.get_annotations())
# Read back into new object
obj2 = create_signal("Test 2")
annotations = read_annotations(filepath)
obj2.set_annotations(annotations)
# Verify
assert obj2.get_annotations() == obj.get_annotations()
assert len(obj2.get_annotations()) == 2
finally:
# Clean up
Path(filepath).unlink(missing_ok=True)
if __name__ == "__main__":
test_write_read_annotations()
test_write_read_annotations_with_object()
print("All tests passed!")
|
