テストコードの追加

Author: yuji38kwmt

tests/test_local_parser.py

@@ -114,6 +114,20 @@ def test_lazy_parse_simple_annotation_zip_by_task(self):
         assert len([e for e in parser_list if e.input_data_id == "c86205d1-bdd4-4110-ae46-194e661d622b"]) == 1
 
 
+    def test_lazy_parse_simple_annotation_dir_by_task(self):
+        zip_path = Path(test_dir / "simple-annotation-v2")
+        task_parser_list = list(annofabapi.parser.lazy_parse_simple_annotation_dir_by_task(zip_path))
+
+        assert len(task_parser_list) == 2
+        assert len([e for e in task_parser_list if e.task_id == "sample_1"]) == 1
+        assert len([e for e in task_parser_list if e.task_id == "sample_0"]) == 1
+
+        task_parser = [e for e in task_parser_list if e.task_id == "sample_1"][0]
+        parser_list = task_parser.parser_list
+        assert len(parser_list) == 2
+        assert len([e for e in parser_list if e.input_data_id == "c6e1c2ec-6c7c-41c6-9639-4244c2ed2839"]) == 1
+        assert len([e for e in parser_list if e.input_data_id == "c86205d1-bdd4-4110-ae46-194e661d622b"]) == 1
+
 class TestFullAnnotation:
     def test_full_annotation_zip(self):
         zip_path = Path(test_dir / "full-annotation.zip")