add

Author: yuji38kwmt

.vscode/cspell.json

@@ -33,6 +33,7 @@
         "pydantic",
         "pydata",
         "pylint",
+        "pytest",
         "relativedelta",
         "timespec",
         "tzlocal",

annofabapi/util/attribute_restrictions.py

@@ -8,10 +8,9 @@
 
         >>> import annofabapi
         >>> from annofabapi.util.annotation_specs import AnnotationSpecsAccessor
-        >>> from annofabapi.util.attribute_restrictions import AttributeFactory
         >>> service = annofabapi.build()
         >>> annotation_specs, _ = service.api.get_annotation_specs("prj1", query_params={"v": "3"})
-        >>> fac = AttributeFactory(AnnotationSpecsAccessor(annotation_specs))
+        >>> fac = AttributeFactory(annotation_specs)
 
         # 「'occluded'チェックボックスがONならば、'note'テキストボックスは空ではない」という制約
         >>> premise_restriction = fac.checkbox(attribute_name="occluded").checked()
@@ -295,11 +294,11 @@ class AttributeFactory:
     属性を生成するためのFactoryクラス
 
     Attributes:
-        accessor: アノテーション仕様のアクセサ
+        annotation_specs: アノテーション仕様(v3)の情報
     """
 
-    def __init__(self, accessor: AnnotationSpecsAccessor) -> None:
-        self.accessor = accessor
+    def __init__(self, annotation_specs: dict[str, Any]) -> None:
+        self.accessor = AnnotationSpecsAccessor(annotation_specs)
 
     def checkbox(self, *, attribute_id: Optional[str] = None, attribute_name: Optional[str] = None) -> Checkbox:
         return Checkbox(self.accessor, attribute_id=attribute_id, attribute_name=attribute_name)

tests/util/test_attribute_restrictions.py

@@ -4,7 +4,7 @@
 import pytest
 
 from annofabapi.util.annotation_specs import AnnotationSpecsAccessor
-from annofabapi.util.attribute_restrictions import AnnotationLink, Checkbox, IntegerTextbox, Selection, StringTextbox, TrackingId
+from annofabapi.util.attribute_restrictions import AnnotationLink, Checkbox, IntegerTextbox, Selection, StringTextbox, TrackingId, AttributeFactory
 
 accessor = AnnotationSpecsAccessor(annotation_specs=json.loads(Path("tests/data/util/attribute_restrictions/annotation_specs.json").read_text()))
 
@@ -161,3 +161,40 @@ def test__implyメソッドの戻りに対してimplyメソッドを実行する
             Checkbox(accessor, attribute_name="occluded").checked().imply(IntegerTextbox(accessor, attribute_name="traffic_lane").equals(2)).imply(
                 StringTextbox(accessor, attribute_name="note").is_not_empty()
             )
+            
+            
+class Test__AttributeFactory:
+    def setup_method(self):
+        self.annotation_specs = json.loads(Path("tests/data/util/attribute_restrictions/annotation_specs.json").read_text())
+        self.factory = AttributeFactory(self.annotation_specs)
+
+    def test__checkbox(self):
+        checkbox = self.factory.checkbox(attribute_name="occluded")
+        assert isinstance(checkbox, Checkbox)
+        assert checkbox.attribute_id == "2517f635-2269-4142-8ef4-16312b4cc9f7"
+
+    def test__string_textbox(self):
+        string_textbox = self.factory.string_textbox(attribute_name="note")
+        assert isinstance(string_textbox, StringTextbox)
+        assert string_textbox.attribute_id == "9b05648d-1e16-4ea2-ab79-48907f5eed00"
+
+    def test__integer_textbox(self):
+        integer_textbox = self.factory.integer_textbox(attribute_name="traffic_lane")
+        assert isinstance(integer_textbox, IntegerTextbox)
+        assert integer_textbox.attribute_id == "ec27de5d-122c-40e7-89bc-5500e37bae6a"
+
+    def test__annotation_link(self):
+        annotation_link = self.factory.annotation_link(attribute_name="link_car")
+        assert isinstance(annotation_link, AnnotationLink)
+        assert annotation_link.attribute_id == "15ba8b9d-4882-40c2-bb31-ed3f68197c2e"
+
+    def test__tracking_id(self):
+        tracking_id = self.factory.tracking_id(attribute_name="tracking")
+        assert isinstance(tracking_id, TrackingId)
+        assert tracking_id.attribute_id == "d349e76d-b59a-44cd-94b4-713a00b2e84d"
+
+    def test__selection(self):
+        selection = self.factory.selection(attribute_name="car_kind")
+        assert isinstance(selection, Selection)
+        assert selection.attribute_id == "cbb0155f-1631-48e1-8fc3-43c5f254b6f2"
+