mirror of
https://github.com/wassname/sloth.git
synced 2026-06-27 18:25:49 +08:00
Martin Baeuml
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
from sloth.annotations.container import *
|
|
|
|
|
|
class MockupContainer:
|
|
pass
|
|
|
|
|
|
def someFileAnnotations(i):
|
|
annotations = [{'type': 'rect', 'x': 10 * i, 'y': '20', 'w': '40', 'h': '60'},
|
|
{'type': 'rect', 'x': '80', 'y': 20 * i, 'w': '40', 'h': '60'}]
|
|
for k in range(i):
|
|
annotations.append({'type': 'point',
|
|
'x': 30 * k,
|
|
'y': 30 * k})
|
|
return annotations
|
|
|
|
|
|
def someAnnotations():
|
|
annotations = []
|
|
for i in range(5):
|
|
ann = {
|
|
'filename': 'file%d.png' % i,
|
|
'type': 'image',
|
|
'annotations': someFileAnnotations(i)
|
|
}
|
|
annotations.append(ann)
|
|
return annotations
|
|
|
|
|
|
def common_container_test(filename, container):
|
|
original_anns = someAnnotations()
|
|
|
|
container.save(original_anns, filename)
|
|
assert container.filename() == filename
|
|
assert os.path.exists(filename)
|
|
|
|
container.clear()
|
|
assert container.filename() is None
|
|
|
|
container.load(filename)
|
|
assert container.filename() == filename
|
|
|
|
|
|
def test_import_callable():
|
|
containers = (('*', 'container_test.MockupContainer'),)
|
|
factory = AnnotationContainerFactory(containers)
|
|
item = factory.create('test')
|
|
assert isinstance(item, MockupContainer)
|
|
|
|
|
|
def test_PickleContainer(tmpdir):
|
|
filename = os.path.join(str(tmpdir), "test_PickleContainer.pickle")
|
|
container = PickleContainer()
|
|
common_container_test(filename, container)
|
|
|
|
|
|
def test_JsonContainer(tmpdir):
|
|
filename = os.path.join(str(tmpdir), "test_JsonContainer.json")
|
|
container = JsonContainer()
|
|
common_container_test(filename, container)
|
|
|
|
|
|
def test_YamlContainer(tmpdir):
|
|
filename = os.path.join(str(tmpdir), "test_YamlContainer.yaml")
|
|
container = YamlContainer()
|
|
common_container_test(filename, container)
|