Merge pull request #2261 from VWS-Python/fix-tests

Avoid SupportsFloat as it doesn't work so well with pydantic

Author: adamtheturtle

docs/source/docker.rst

@@ -136,7 +136,7 @@ VWS container
 
    The number of seconds to process each image for.
 
-   Default: ``2``
+   Default: ``2.0``
 
 .. envvar:: DUPLICATES_IMAGE_MATCHER
 

src/mock_vws/_flask_server/vws.py

@@ -12,7 +12,6 @@
 import uuid
 from enum import StrEnum, auto
 from http import HTTPMethod, HTTPStatus
-from typing import SupportsFloat
 
 import requests
 from flask import Flask, Response, request
@@ -67,7 +66,7 @@ class VWSSettings(BaseSettings):
     """Settings for the VWS Flask app."""
 
     target_manager_base_url: str
-    processing_time_seconds: SupportsFloat = 2
+    processing_time_seconds: float = 2.0
     vws_host: str = ""
     duplicates_image_matcher: _ImageMatcherChoice = (
         _ImageMatcherChoice.STRUCTURAL_SIMILARITY

src/mock_vws/_requests_mock_server/decorators.py

@@ -4,7 +4,7 @@
 
 import re
 from contextlib import ContextDecorator
-from typing import Literal, Self, SupportsFloat
+from typing import Literal, Self
 from urllib.parse import urljoin, urlparse
 
 import requests
@@ -39,7 +39,7 @@ def __init__(
         base_vwq_url: str = "https://cloudreco.vuforia.com",
         duplicate_match_checker: ImageMatcher = _STRUCTURAL_SIMILARITY_MATCHER,
         query_match_checker: ImageMatcher = _STRUCTURAL_SIMILARITY_MATCHER,
-        processing_time_seconds: SupportsFloat = 2,
+        processing_time_seconds: float = 2.0,
         target_tracking_rater: TargetTrackingRater = _BRISQUE_TRACKING_RATER,
         *,
         real_http: bool = False,

src/mock_vws/_requests_mock_server/mock_web_services_api.py

@@ -12,7 +12,7 @@
 import uuid
 from collections.abc import Callable
 from http import HTTPMethod, HTTPStatus
-from typing import TYPE_CHECKING, SupportsFloat
+from typing import TYPE_CHECKING
 from zoneinfo import ZoneInfo
 
 from mock_vws._constants import ResultCodes, TargetStatuses
@@ -103,7 +103,7 @@ def __init__(
         self,
         *,
         target_manager: TargetManager,
-        processing_time_seconds: SupportsFloat,
+        processing_time_seconds: float,
         duplicate_match_checker: ImageMatcher,
         target_tracking_rater: TargetTrackingRater,
     ) -> None:

src/mock_vws/target.py

@@ -8,7 +8,7 @@
 import statistics
 import uuid
 from dataclasses import dataclass, field
-from typing import Self, SupportsFloat, TypedDict
+from typing import Self, TypedDict
 from zoneinfo import ZoneInfo
 
 from PIL import Image, ImageStat
@@ -29,7 +29,7 @@ class TargetDict(TypedDict):
     width: float
     image_base64: str
     active_flag: bool
-    processing_time_seconds: SupportsFloat
+    processing_time_seconds: float
     application_metadata: str | None
     target_id: str
     last_modified_date: str
@@ -64,7 +64,7 @@ class Target:
     application_metadata: str | None
     image_value: bytes
     name: str
-    processing_time_seconds: SupportsFloat
+    processing_time_seconds: float
     width: float
     target_tracking_rater: TargetTrackingRater = field(compare=False)
     current_month_recos: int = 0

tests/mock_vws/test_flask_app_usage.py

@@ -89,7 +89,7 @@ def test_custom(
         """
         It is possible to set a custom processing time.
         """
-        seconds = 5
+        seconds = 5.0
         monkeypatch.setenv(
             name="PROCESSING_TIME_SECONDS",
             value=str(seconds),