From 638c3db45d1e6affd8e7540a549f1c5af5892dae Mon Sep 17 00:00:00 2001 From: Martin Baeuml Date: Wed, 19 Mar 2014 14:15:06 +0100 Subject: [PATCH] pep8 changes --- tests/container_test.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/container_test.py b/tests/container_test.py index e39e1c1..819668d 100644 --- a/tests/container_test.py +++ b/tests/container_test.py @@ -1,8 +1,10 @@ from sloth.annotations.container import * + class MockupContainer: pass + def someFileAnnotations(i): annotations = [] annotations.append({'type': 'rect', @@ -21,6 +23,7 @@ def someFileAnnotations(i): 'y': 30 * k}) return annotations + def someAnnotations(): annotations = [] for i in range(5): @@ -32,6 +35,7 @@ def someAnnotations(): annotations.append(file) return annotations + def common_container_test(filename, container): original_anns = someAnnotations() @@ -50,22 +54,26 @@ def common_container_test(filename, container): assert container.filename() == filename assert len(container.annotations()) == len(original_anns) + 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()