mocra 0.1.0__tar.gz

mocra-0.1.0/PKG-INFO

@@ -0,0 +1,134 @@
+Metadata-Version: 2.4
+Name: mocra
+Version: 0.1.0
+Summary: Python SDK for Mocra's Observe API - observability for video generation workflows
+Author: Mocra
+License: MIT
+Project-URL: Documentation, https://docs.mocra.io
+Project-URL: Homepage, https://mocra.io
+Keywords: mocra,video,observability,quality,ai
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pydantic>=2.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Requires-Dist: ruff>=0.8.0; extra == "dev"
+Requires-Dist: pre-commit>=3.0.0; extra == "dev"
+
+# Mocra Python SDK
+
+Python SDK for [Mocra's Observe API](https://docs.mocra.io/) — observability for video generation workflows. The API mirrors the [TypeScript SDK](https://github.com/Mocra-AI/Mocra) so you can switch between languages easily.
+
+## Installation
+
+```bash
+pip install -e .
+```
+
+## Requirements
+
+- Python 3.10+
+- [httpx](https://www.python-httpx.org/)
+- [pydantic](https://docs.pydantic.dev/) v2
+
+## Quick Start
+
+```python
+from mocra import VideoObservabilityApi, ExtraCriterion
+
+api = VideoObservabilityApi("YOUR_API_KEY")
+result = api.score_video(
+    "https://example.org/video.mp4",
+    extra_criteria=[
+        ExtraCriterion(
+            criterion_name="Blur",
+            criterion_description="The video should not be blurry",
+        ),
+    ],
+    ignore_criteria=["UNNATURAL PHYSICS"],
+)
+
+print(f"Overall score: {result.severity}")
+for criterion in result.criteria:
+    print(f"  {criterion.name}: {criterion.score}")
+```
+
+## API Mapping (TypeScript ↔ Python)
+
+| TypeScript | Python |
+|------------|--------|
+| `VideoObservabilityApi` | `VideoObservabilityApi` |
+| `scoreVideo(videoUrl, extraCriteria?, ignoreCriteria?)` | `score_video(video_url, extra_criteria?, ignore_criteria?)` |
+| `ExtraCriterion { criterionName, criterionDescription }` | `ExtraCriterion(criterion_name=..., criterion_description=...)` |
+| `ignoreCriteria: ["UNNATURAL PHYSICS", ...]` | `ignore_criteria=["UNNATURAL PHYSICS"]` or `[DefaultCriterion.UNNATURAL_PHYSICS]` |
+| `ScoreMap { severity, criteria }` | `ObserveResponse` |
+
+**Default criteria** (for `ignore_criteria`): `UNNATURAL PHYSICS`, `MORPHING`, `FLICKERING`, `ARTIFACTING`, `TEXT ISSUES`
+
+## API Reference
+
+### VideoObservabilityApi
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `api_key` | `str` | Your Mocra API key ([get one](https://docs.mocra.io/quickstart)) |
+| `base_url` | `str` | Override API URL (default: `https://api.mocra.io`) |
+| `timeout` | `float` | Request timeout in seconds |
+| `http_client` | `httpx.Client` | Optional pre-configured client |
+
+### score_video()
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `video_url` | `str` | URL of the video to analyze |
+| `extra_criteria` | `list[ExtraCriterion]` | Extra criteria (default: `[]`) |
+| `ignore_criteria` | `list[str \| DefaultCriterion]` | Default criteria to exclude (default: `[]`) |
+
+**Returns:** `{ severity, criteria: [{ name, score }] }` (1–100 scale)
+
+## Context Manager
+
+```python
+with VideoObservabilityApi("YOUR_API_KEY") as api:
+    result = api.score_video("https://example.org/video.mp4")
+```
+
+## Development
+
+### Setup
+
+```bash
+pip install -e ".[dev]"
+pre-commit install  # Run lint on git commit
+```
+
+### Lint
+
+```bash
+ruff check mocra tests
+```
+
+### Test
+
+```bash
+pytest tests/ -v
+```
+
+### Publishing to PyPI
+
+1. Create an API token at [pypi.org/manage/account/token](https://pypi.org/manage/account/token)
+2. Add `PYPI_API_TOKEN` as a repository secret in GitHub
+3. Create a release or run the workflow manually — the publish workflow runs on release publish and `workflow_dispatch`
+
+## License
+
+MIT

mocra-0.1.0/README.md

@@ -0,0 +1,108 @@
+# Mocra Python SDK
+
+Python SDK for [Mocra's Observe API](https://docs.mocra.io/) — observability for video generation workflows. The API mirrors the [TypeScript SDK](https://github.com/Mocra-AI/Mocra) so you can switch between languages easily.
+
+## Installation
+
+```bash
+pip install -e .
+```
+
+## Requirements
+
+- Python 3.10+
+- [httpx](https://www.python-httpx.org/)
+- [pydantic](https://docs.pydantic.dev/) v2
+
+## Quick Start
+
+```python
+from mocra import VideoObservabilityApi, ExtraCriterion
+
+api = VideoObservabilityApi("YOUR_API_KEY")
+result = api.score_video(
+    "https://example.org/video.mp4",
+    extra_criteria=[
+        ExtraCriterion(
+            criterion_name="Blur",
+            criterion_description="The video should not be blurry",
+        ),
+    ],
+    ignore_criteria=["UNNATURAL PHYSICS"],
+)
+
+print(f"Overall score: {result.severity}")
+for criterion in result.criteria:
+    print(f"  {criterion.name}: {criterion.score}")
+```
+
+## API Mapping (TypeScript ↔ Python)
+
+| TypeScript | Python |
+|------------|--------|
+| `VideoObservabilityApi` | `VideoObservabilityApi` |
+| `scoreVideo(videoUrl, extraCriteria?, ignoreCriteria?)` | `score_video(video_url, extra_criteria?, ignore_criteria?)` |
+| `ExtraCriterion { criterionName, criterionDescription }` | `ExtraCriterion(criterion_name=..., criterion_description=...)` |
+| `ignoreCriteria: ["UNNATURAL PHYSICS", ...]` | `ignore_criteria=["UNNATURAL PHYSICS"]` or `[DefaultCriterion.UNNATURAL_PHYSICS]` |
+| `ScoreMap { severity, criteria }` | `ObserveResponse` |
+
+**Default criteria** (for `ignore_criteria`): `UNNATURAL PHYSICS`, `MORPHING`, `FLICKERING`, `ARTIFACTING`, `TEXT ISSUES`
+
+## API Reference
+
+### VideoObservabilityApi
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `api_key` | `str` | Your Mocra API key ([get one](https://docs.mocra.io/quickstart)) |
+| `base_url` | `str` | Override API URL (default: `https://api.mocra.io`) |
+| `timeout` | `float` | Request timeout in seconds |
+| `http_client` | `httpx.Client` | Optional pre-configured client |
+
+### score_video()
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `video_url` | `str` | URL of the video to analyze |
+| `extra_criteria` | `list[ExtraCriterion]` | Extra criteria (default: `[]`) |
+| `ignore_criteria` | `list[str \| DefaultCriterion]` | Default criteria to exclude (default: `[]`) |
+
+**Returns:** `{ severity, criteria: [{ name, score }] }` (1–100 scale)
+
+## Context Manager
+
+```python
+with VideoObservabilityApi("YOUR_API_KEY") as api:
+    result = api.score_video("https://example.org/video.mp4")
+```
+
+## Development
+
+### Setup
+
+```bash
+pip install -e ".[dev]"
+pre-commit install  # Run lint on git commit
+```
+
+### Lint
+
+```bash
+ruff check mocra tests
+```
+
+### Test
+
+```bash
+pytest tests/ -v
+```
+
+### Publishing to PyPI
+
+1. Create an API token at [pypi.org/manage/account/token](https://pypi.org/manage/account/token)
+2. Add `PYPI_API_TOKEN` as a repository secret in GitHub
+3. Create a release or run the workflow manually — the publish workflow runs on release publish and `workflow_dispatch`
+
+## License
+
+MIT

mocra-0.1.0/mocra/__init__.py

@@ -0,0 +1,35 @@
+"""
+Mocra Python SDK - Observability for video generation workflows.
+
+API mirrors the TypeScript SDK: https://github.com/Mocra-AI/Mocra
+
+    from mocra import VideoObservabilityApi, ExtraCriterion
+
+    api = VideoObservabilityApi("your-api-key")
+    result = api.score_video(
+        "https://example.org/video.mp4",
+        extra_criteria=[
+            ExtraCriterion(criterion_name="Blur", criterion_description="..."),
+        ],
+        ignore_criteria=["UNNATURAL PHYSICS"],
+    )
+"""
+
+from mocra.client import MocraError, VideoObservabilityApi
+from mocra.models import (
+    CriterionScore,
+    DefaultCriterion,
+    ExtraCriterion,
+    ObserveResponse,
+)
+
+__all__ = [
+    "VideoObservabilityApi",
+    "MocraError",
+    "ExtraCriterion",
+    "DefaultCriterion",
+    "ObserveResponse",
+    "CriterionScore",
+]
+
+__version__ = "0.1.0"

mocra-0.1.0/mocra/client.py

@@ -0,0 +1,150 @@
+"""
+HTTP client for the Mocra Observe API.
+
+API mirrors the TypeScript SDK: VideoObservabilityApi(api_key).score_video(...)
+See https://github.com/Mocra-AI/Mocra
+"""
+
+import httpx
+
+from mocra.models import (
+    DefaultCriterion,
+    ExtraCriterion,
+    ObserveRequest,
+    ObserveResponse,
+)
+
+MOCRA_API_DOMAIN = "https://api.mocra.io"
+OBSERVE_PATH = "/observe"
+
+DEFAULT_HEADERS = {
+    "Content-Type": "application/json",
+    "Accept": "application/json",
+}
+
+
+def _normalize_ignore_criteria(
+    criteria: list[str | DefaultCriterion],
+) -> list[str]:
+    """Convert ignore_criteria to list of strings for API payload."""
+    return [
+        c.value if isinstance(c, DefaultCriterion) else str(c)
+        for c in criteria
+    ]
+
+
+def _parse_error_message(response: httpx.Response) -> str:
+    """Extract error message from failed API response."""
+    try:
+        body = response.json()
+        return body.get("message", response.text)
+    except Exception:
+        return response.text
+
+
+class MocraError(Exception):
+    """Raised when the Mocra API returns an error."""
+
+    def __init__(self, message: str, status_code: int | None = None):
+        self.message = message
+        self.status_code = status_code
+        super().__init__(f"[{status_code}] {message}" if status_code else message)
+
+
+class VideoObservabilityApi:
+    """
+    Video observability API. Matches TypeScript VideoObservabilityApi.
+
+    Example (mirrors TS SDK):
+        >>> api = VideoObservabilityApi("your-api-key")
+        >>> result = api.score_video(
+        ...     "http://example.org/video.mp4",
+        ...     extra_criteria=[
+        ...         ExtraCriterion(
+        ...             criterion_name="Blur",
+        ...             criterion_description="The video should not be blurry",
+        ...         ),
+        ...     ],
+        ...     ignore_criteria=["UNNATURAL PHYSICS"],
+        ... )
+        >>> print(result.severity, result.criteria)
+    """
+
+    def __init__(
+        self,
+        api_key: str,
+        base_url: str | None = None,
+        timeout: float = 60.0,
+        http_client: httpx.Client | None = None,
+    ):
+        """
+        Initialize the API client. Matches TS: VideoObservabilityApi(apiKey).
+
+        Args:
+            api_key: Your Mocra API key.
+            base_url: Override API base URL (default: https://api.mocra.io).
+            timeout: Request timeout in seconds.
+            http_client: Optional pre-configured httpx client.
+        """
+        self._api_key = api_key
+        base = (base_url or MOCRA_API_DOMAIN).rstrip("/") + "/"
+        self._timeout = timeout
+        headers = {"Authorization": f"Bearer {api_key}", **DEFAULT_HEADERS}
+        self._client = http_client or httpx.Client(
+            base_url=base,
+            headers=headers,
+            timeout=timeout,
+        )
+        self._owns_client = http_client is None
+
+    def score_video(
+        self,
+        video_url: str,
+        extra_criteria: list[ExtraCriterion] | None = None,
+        ignore_criteria: list[str | DefaultCriterion] | None = None,
+    ) -> ObserveResponse:
+        """
+        Score a video (TS: scoreVideo(videoUrl, extraCriteria?, ignoreCriteria?)).
+
+        Args:
+            video_url: URL of the video to analyze.
+            extra_criteria: Extra criteria to analyze (default: []).
+            ignore_criteria: Default criteria to exclude, e.g. ["UNNATURAL PHYSICS"]
+                or [DefaultCriterion.UNNATURAL_PHYSICS] (default: []).
+
+        Returns:
+            { severity, criteria: [{ name, score }] }
+
+        Raises:
+            MocraError: API returned an error.
+            httpx.HTTPError: Network error.
+        """
+        extra = extra_criteria or []
+        ignore_strs = _normalize_ignore_criteria(ignore_criteria or [])
+        request = ObserveRequest(
+            video_url=video_url,
+            custom_criteria=extra,
+            remove_criteria=ignore_strs,
+        )
+        return self._request(request)
+
+    def _request(self, request: ObserveRequest) -> ObserveResponse:
+        payload = request.model_dump(by_alias=True, mode="json")
+        response = self._client.post(OBSERVE_PATH, json=payload)
+
+        if response.status_code == 200:
+            return ObserveResponse.model_validate(response.json())
+
+        message = _parse_error_message(response)
+        raise MocraError(message=message, status_code=response.status_code)
+
+    def close(self) -> None:
+        """Close the HTTP client and release resources."""
+        if self._owns_client:
+            self._client.close()
+
+    def __enter__(self) -> "VideoObservabilityApi":
+        return self
+
+    def __exit__(self, *args: object) -> None:
+        self.close()

mocra-0.1.0/mocra/models.py

@@ -0,0 +1,66 @@
+"""
+Data models for the Mocra Observe API.
+"""
+
+from enum import Enum
+
+from pydantic import BaseModel, Field
+
+
+class DefaultCriterion(str, Enum):
+    """Built-in criteria that can be removed from analysis."""
+
+    UNNATURAL_PHYSICS = "UNNATURAL PHYSICS"
+    MORPHING = "MORPHING"
+    FLICKERING = "FLICKERING"
+    ARTIFACTING = "ARTIFACTING"
+    TEXT_ISSUES = "TEXT ISSUES"
+
+
+class ExtraCriterion(BaseModel):
+    """
+    Extra criterion to add to the analysis.
+    Matches TypeScript ExtraCriterion: { criterionName, criterionDescription }
+    """
+
+    criterion_name: str = Field(..., alias="criterionName")
+    criterion_description: str = Field(..., alias="criterionDescription")
+
+    model_config = {"populate_by_name": True}
+
+
+class ObserveRequest(BaseModel):
+    """Request body for the /observe endpoint."""
+
+    video_url: str = Field(..., alias="videoUrl")
+    custom_criteria: list[ExtraCriterion] = Field(
+        default_factory=list, alias="customCriteria"
+    )
+    remove_criteria: list[str] = Field(
+        default_factory=list, alias="removeCriteria"
+    )
+
+    model_config = {"populate_by_name": True}
+
+
+class CriterionScore(BaseModel):
+    """Per-criterion score in the response."""
+
+    name: str
+    score: float = Field(..., ge=1, le=100)
+
+
+class ObserveResponse(BaseModel):
+    """
+    Response from the /observe endpoint.
+    Matches TypeScript ScoreMap: { severity, criteria: [{ name, score }] }
+    """
+
+    severity: float = Field(..., ge=1, le=100)
+    criteria: list[CriterionScore]
+
+
+class ErrorResponse(BaseModel):
+    """Error response from the API."""
+
+    message: str

mocra-0.1.0/mocra.egg-info/PKG-INFO

@@ -0,0 +1,134 @@
+Metadata-Version: 2.4
+Name: mocra
+Version: 0.1.0
+Summary: Python SDK for Mocra's Observe API - observability for video generation workflows
+Author: Mocra
+License: MIT
+Project-URL: Documentation, https://docs.mocra.io
+Project-URL: Homepage, https://mocra.io
+Keywords: mocra,video,observability,quality,ai
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pydantic>=2.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Requires-Dist: ruff>=0.8.0; extra == "dev"
+Requires-Dist: pre-commit>=3.0.0; extra == "dev"
+
+# Mocra Python SDK
+
+Python SDK for [Mocra's Observe API](https://docs.mocra.io/) — observability for video generation workflows. The API mirrors the [TypeScript SDK](https://github.com/Mocra-AI/Mocra) so you can switch between languages easily.
+
+## Installation
+
+```bash
+pip install -e .
+```
+
+## Requirements
+
+- Python 3.10+
+- [httpx](https://www.python-httpx.org/)
+- [pydantic](https://docs.pydantic.dev/) v2
+
+## Quick Start
+
+```python
+from mocra import VideoObservabilityApi, ExtraCriterion
+
+api = VideoObservabilityApi("YOUR_API_KEY")
+result = api.score_video(
+    "https://example.org/video.mp4",
+    extra_criteria=[
+        ExtraCriterion(
+            criterion_name="Blur",
+            criterion_description="The video should not be blurry",
+        ),
+    ],
+    ignore_criteria=["UNNATURAL PHYSICS"],
+)
+
+print(f"Overall score: {result.severity}")
+for criterion in result.criteria:
+    print(f"  {criterion.name}: {criterion.score}")
+```
+
+## API Mapping (TypeScript ↔ Python)
+
+| TypeScript | Python |
+|------------|--------|
+| `VideoObservabilityApi` | `VideoObservabilityApi` |
+| `scoreVideo(videoUrl, extraCriteria?, ignoreCriteria?)` | `score_video(video_url, extra_criteria?, ignore_criteria?)` |
+| `ExtraCriterion { criterionName, criterionDescription }` | `ExtraCriterion(criterion_name=..., criterion_description=...)` |
+| `ignoreCriteria: ["UNNATURAL PHYSICS", ...]` | `ignore_criteria=["UNNATURAL PHYSICS"]` or `[DefaultCriterion.UNNATURAL_PHYSICS]` |
+| `ScoreMap { severity, criteria }` | `ObserveResponse` |
+
+**Default criteria** (for `ignore_criteria`): `UNNATURAL PHYSICS`, `MORPHING`, `FLICKERING`, `ARTIFACTING`, `TEXT ISSUES`
+
+## API Reference
+
+### VideoObservabilityApi
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `api_key` | `str` | Your Mocra API key ([get one](https://docs.mocra.io/quickstart)) |
+| `base_url` | `str` | Override API URL (default: `https://api.mocra.io`) |
+| `timeout` | `float` | Request timeout in seconds |
+| `http_client` | `httpx.Client` | Optional pre-configured client |
+
+### score_video()
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `video_url` | `str` | URL of the video to analyze |
+| `extra_criteria` | `list[ExtraCriterion]` | Extra criteria (default: `[]`) |
+| `ignore_criteria` | `list[str \| DefaultCriterion]` | Default criteria to exclude (default: `[]`) |
+
+**Returns:** `{ severity, criteria: [{ name, score }] }` (1–100 scale)
+
+## Context Manager
+
+```python
+with VideoObservabilityApi("YOUR_API_KEY") as api:
+    result = api.score_video("https://example.org/video.mp4")
+```
+
+## Development
+
+### Setup
+
+```bash
+pip install -e ".[dev]"
+pre-commit install  # Run lint on git commit
+```
+
+### Lint
+
+```bash
+ruff check mocra tests
+```
+
+### Test
+
+```bash
+pytest tests/ -v
+```
+
+### Publishing to PyPI
+
+1. Create an API token at [pypi.org/manage/account/token](https://pypi.org/manage/account/token)
+2. Add `PYPI_API_TOKEN` as a repository secret in GitHub
+3. Create a release or run the workflow manually — the publish workflow runs on release publish and `workflow_dispatch`
+
+## License
+
+MIT

mocra-0.1.0/mocra.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+mocra/__init__.py
+mocra/client.py
+mocra/models.py
+mocra.egg-info/PKG-INFO
+mocra.egg-info/SOURCES.txt
+mocra.egg-info/dependency_links.txt
+mocra.egg-info/requires.txt
+mocra.egg-info/top_level.txt
+tests/test_client.py
+tests/test_models.py

mocra-0.1.0/mocra.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

mocra-0.1.0/mocra.egg-info/requires.txt

@@ -0,0 +1,8 @@
+httpx>=0.24.0
+pydantic>=2.0.0
+
+[dev]
+pytest>=7.0
+pytest-cov>=4.0
+ruff>=0.8.0
+pre-commit>=3.0.0

mocra-0.1.0/mocra.egg-info/top_level.txt

@@ -0,0 +1 @@
+mocra

mocra-0.1.0/pyproject.toml

@@ -0,0 +1,68 @@
+[build-system]
+requires = ["setuptools>=61", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "mocra"
+version = "0.1.0"
+description = "Python SDK for Mocra's Observe API - observability for video generation workflows"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.10"
+authors = [
+    { name = "Mocra" }
+]
+keywords = ["mocra", "video", "observability", "quality", "ai"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+    "httpx>=0.24.0",
+    "pydantic>=2.0.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0",
+    "pytest-cov>=4.0",
+    "ruff>=0.8.0",
+    "pre-commit>=3.0.0",
+]
+
+[project.urls]
+Documentation = "https://docs.mocra.io"
+Homepage = "https://mocra.io"
+
+[tool.setuptools.packages.find]
+include = ["mocra*"]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 88
+src = ["mocra", "tests"]
+
+[tool.ruff.lint]
+select = [
+    "E",      # pycodestyle errors
+    "W",      # pycodestyle warnings
+    "F",      # Pyflakes
+    "I",      # isort
+    "B",      # flake8-bugbear
+    "C4",     # flake8-comprehensions
+    "UP",     # pyupgrade
+]
+ignore = []
+
+[tool.ruff.lint.isort]
+known-first-party = ["mocra"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+pythonpath = ["."]
+addopts = "-v --tb=short"

mocra-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

mocra-0.1.0/tests/test_client.py

@@ -0,0 +1,94 @@
+"""Tests for VideoObservabilityApi client."""
+
+import httpx
+import pytest
+
+from mocra import ExtraCriterion, MocraError, VideoObservabilityApi
+from tests.conftest import SAMPLE_VIDEO_URL, create_mock_api
+
+
+def test_video_observability_api_constructor_accepts_api_key(api_key: str) -> None:
+    """Should accept API key without checking validity (matches TS SDK)."""
+    api = VideoObservabilityApi(api_key)
+    assert api._api_key == api_key
+
+
+def test_score_video_success(mock_httpx_client: VideoObservabilityApi) -> None:
+    """Should return parsed response on 200."""
+    result = mock_httpx_client.score_video(SAMPLE_VIDEO_URL)
+    assert result.severity == 85.5
+    assert len(result.criteria) == 2
+    assert result.criteria[0].name == "UNNATURAL PHYSICS"
+    assert result.criteria[0].score == 90.0
+    assert result.criteria[1].name == "Blur"
+    assert result.criteria[1].score == 81.0
+
+
+def test_score_video_with_extra_criteria(
+    mock_httpx_client: VideoObservabilityApi,
+) -> None:
+    """Should accept extra_criteria and pass to API."""
+    result = mock_httpx_client.score_video(
+        SAMPLE_VIDEO_URL,
+        extra_criteria=[
+            ExtraCriterion(
+                criterion_name="Blur",
+                criterion_description="Video should not be blurry",
+            ),
+        ],
+    )
+    assert result.severity == 85.5
+
+
+def test_score_video_with_ignore_criteria(
+    mock_httpx_client: VideoObservabilityApi,
+) -> None:
+    """Should accept ignore_criteria as strings."""
+    result = mock_httpx_client.score_video(
+        SAMPLE_VIDEO_URL,
+        ignore_criteria=["UNNATURAL PHYSICS", "MORPHING"],
+    )
+    assert result.severity == 85.5
+
+
+def test_score_video_with_default_criterion_enum(
+    mock_httpx_client: VideoObservabilityApi,
+) -> None:
+    """Should accept DefaultCriterion enum in ignore_criteria."""
+    from mocra.models import DefaultCriterion
+
+    result = mock_httpx_client.score_video(
+        SAMPLE_VIDEO_URL,
+        ignore_criteria=[DefaultCriterion.UNNATURAL_PHYSICS],
+    )
+    assert result.severity == 85.5
+
+
+def test_score_video_raises_on_error() -> None:
+    """Should raise MocraError on non-200 response."""
+    response = httpx.Response(401, json={"message": "Invalid API key"})
+    api = create_mock_api(response, api_key="bad-key")
+
+    with pytest.raises(MocraError) as exc_info:
+        api.score_video(SAMPLE_VIDEO_URL)
+
+    assert exc_info.value.status_code == 401
+    assert "Invalid API key" in str(exc_info.value)
+
+
+def test_score_video_raises_on_400() -> None:
+    """Should raise MocraError with message on 400."""
+    response = httpx.Response(400, json={"message": "Invalid video URL"})
+    api = create_mock_api(response)
+
+    with pytest.raises(MocraError) as exc_info:
+        api.score_video("not-a-valid-url")
+    assert exc_info.value.message == "Invalid video URL"
+
+
+def test_context_manager_closes_client(api_key: str) -> None:
+    """Should close client when used as context manager."""
+    with VideoObservabilityApi(api_key) as api:
+        assert api._client is not None
+    # Client should be closed (we can't easily assert without internal access,
+    # but at least it shouldn't raise)

mocra-0.1.0/tests/test_models.py

@@ -0,0 +1,44 @@
+"""Tests for mocra models."""
+
+import pytest
+from pydantic import ValidationError
+
+from mocra.models import (
+    CriterionScore,
+    DefaultCriterion,
+    ExtraCriterion,
+    ObserveResponse,
+)
+
+
+def test_extra_criterion_serialization() -> None:
+    """ExtraCriterion should serialize with camelCase aliases."""
+    c = ExtraCriterion(
+        criterion_name="Blur",
+        criterion_description="No blur allowed",
+    )
+    d = c.model_dump(by_alias=True)
+    assert d["criterionName"] == "Blur"
+    assert d["criterionDescription"] == "No blur allowed"
+
+
+def test_default_criterion_values() -> None:
+    """DefaultCriterion should have correct string values."""
+    assert DefaultCriterion.UNNATURAL_PHYSICS.value == "UNNATURAL PHYSICS"
+    assert DefaultCriterion.TEXT_ISSUES.value == "TEXT ISSUES"
+
+
+def test_observe_response_validation() -> None:
+    """ObserveResponse should validate severity and criteria."""
+    r = ObserveResponse(severity=75.5, criteria=[CriterionScore(name="A", score=80)])
+    assert r.severity == 75.5
+    assert r.criteria[0].name == "A"
+    assert r.criteria[0].score == 80
+
+
+def test_observe_response_rejects_invalid_severity() -> None:
+    """ObserveResponse should reject severity outside 1-100."""
+    with pytest.raises(ValidationError):
+        ObserveResponse(severity=0, criteria=[])
+    with pytest.raises(ValidationError):
+        ObserveResponse(severity=101, criteria=[])