🔧 add test config

codegen/__init__.py

@@ -162,6 +162,12 @@ def build_latest_version(
 ):
     logger.info("Start generating latest version...")
 
+    # build pkg
+    logger.info("Building latest __init__.py...")
+    init_template = env.get_template("__init__.py.jinja")
+    init_path = dir / "__init__.py"
+    init_path.write_text(init_template.render())
+
     # build models
     logger.info("Building latest models...")
     latest_template = env.get_template("latest/models.py.jinja")
@@ -296,6 +302,11 @@ def build():
         version_path = config.output_dir / version_module
         version_path.mkdir(parents=True, exist_ok=True)
 
+        # generate __init__.py
+        init_template = env.get_template("__init__.py.jinja")
+        init_path = version_path / "__init__.py"
+        init_path.write_text(init_template.render())
+
         # generate models
         model_path = version_path / "models"
         model_path.mkdir(parents=True, exist_ok=True)

codegen/templates/__init__.py.jinja

@@ -0,0 +1,3 @@
+{% from "header.jinja" import header %}
+
+"""{{ header() }}"""

githubkit/compat.py

@@ -29,7 +29,7 @@ def __get_validators__(cls) -> Generator[Callable[..., Any], None, None]:
     CVC = TypeVar("CVC", bound=CustomValidationClass)
 
 
-if PYDANTIC_V2:
+if PYDANTIC_V2:  # pragma: pydantic-v2
     from pydantic_core import CoreSchema, core_schema
     from pydantic import (
         BaseModel,
@@ -80,7 +80,7 @@ def custom_validation(class_: Type["CVC"]) -> Type["CVC"]:
         )
         return class_
 
-else:
+else:  # pragma: pydantic-v1
     from pydantic import Extra, BaseModel, parse_obj_as, parse_raw_as, root_validator
 
     class GitHubModel(BaseModel):

githubkit/typing.py

@@ -45,7 +45,7 @@
 ]
 RequestFiles: TypeAlias = Union[Dict[str, FileTypes], List[Tuple[str, FileTypes]]]
 
-if PYDANTIC_V2:
+if PYDANTIC_V2:  # pragma: pydantic-v2
     from pydantic import AfterValidator
     from pydantic_core import PydanticCustomError
 
@@ -59,7 +59,7 @@ def _validate_unique_list(value: List[H]) -> List[H]:
         AfterValidator(_validate_unique_list),
         Field(json_schema_extra={"uniqueItems": True}),
     ]
-else:
+else:  # pragma: pydantic-v1
     UniqueList: TypeAlias = Annotated[List[H], Field(unique_items=True)]  # type: ignore
 
 # if the property is not required, we allow it to have the value null.

githubkit/utils.py

@@ -7,7 +7,7 @@
 
 from .compat import PYDANTIC_V2, custom_validation, type_validate_python
 
-if PYDANTIC_V2:
+if PYDANTIC_V2:  # pragma: pydantic-v2
     from pydantic_core import core_schema
     from pydantic import GetCoreSchemaHandler
 
@@ -94,7 +94,7 @@ def __init__(self, type_: Type[T], discriminator: str, tag: str) -> None:
     def _validate(self, value: Any) -> T:
         return type_validate_python(self.type_, value)
 
-    if PYDANTIC_V2:
+    if PYDANTIC_V2:  # pragma: pydantic-v2
 
         def __get_pydantic_core_schema__(
             self, _source_type: Any, _handler: "GetCoreSchemaHandler"

githubkit/versions/latest/__init__.py

@@ -0,0 +1,8 @@
+"""DO NOT EDIT THIS FILE!
+
+This file is automatically @generated by githubkit using the follow command:
+
+python -m codegen && isort . && black .
+
+See https://github.com/github/rest-api-description for more information.
+"""

githubkit/versions/v2022_11_28/__init__.py

@@ -0,0 +1,8 @@
+"""DO NOT EDIT THIS FILE!
+
+This file is automatically @generated by githubkit using the follow command:
+
+python -m codegen && isort . && black .
+
+See https://github.com/github/rest-api-description for more information.
+"""

pyproject.toml

@@ -36,6 +36,7 @@ anyio = "*"
 pytest = "^7.4.3"
 pytest-cov = "^4.1.0"
 pytest-xdist = "^3.5.0"
+coverage-conditional-plugin = "^0.9.0"
 
 [tool.poetry.extras]
 jwt = ["PyJWT"]
@@ -48,6 +49,28 @@ all = ["PyJWT", "anyio"]
 addopts = "--cov=githubkit --cov-append --cov-report=term-missing"
 filterwarnings = ["error", "ignore::DeprecationWarning"]
 
+[tool.coverage.run]
+plugins = ["coverage_conditional_plugin"]
+
+[tool.coverage.report]
+exclude_lines = [
+  "pragma: no cover",
+  "def __repr__",
+  "def __str__",
+  "@(typing\\.)?overload",
+  "if (typing\\.)?TYPE_CHECKING( is True)?:",
+  "@(abc\\.)?abstractmethod",
+  "raise NotImplementedError",
+  "warnings\\.warn",
+  "^\\.\\.\\.$",
+  "pass",
+  "if __name__ == .__main__.:",
+]
+
+[tool.coverage.coverage_conditional_plugin.rules]
+pydantic-v2 = "package_version('pydantic') < (2,)"
+pydantic-v1 = "package_version('pydantic') >= (2,)"
+
 [tool.black]
 line-length = 88
 target-version = ["py38", "py39", "py310", "py311"]