UncountablePythonSDK 0.0.179__tar.gz → 0.0.180__tar.gz

{uncountablepythonsdk-0.0.179 → uncountablepythonsdk-0.0.180}/.github/workflows/documentation.yml

@@ -10,7 +10,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7
-      - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v6
+      - uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7
       - uses: extractions/setup-crate@0551596312d4008a6dfbc4d3c38fa20eaef46d2d # v2
         with:
           repo: casey/just@1.50.0

{uncountablepythonsdk-0.0.179/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.180}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.179
+Version: 0.0.180
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.179 → uncountablepythonsdk-0.0.180/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.179
+Version: 0.0.180
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.179 → uncountablepythonsdk-0.0.180}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -41,6 +41,7 @@ examples/oauth.py
 examples/set_input_pdf_attribute.py
 examples/set_recipe_metadata_file.py
 examples/set_recipe_output_file_sdk.py
+examples/stream_dataset_measurements.py
 examples/upload_files.py
 examples/integration-server/pyproject.toml
 examples/integration-server/jobs/materials_auto/concurrent_cron.py
@@ -84,6 +85,7 @@ pkgs/serialization/serial_class.py
 pkgs/serialization/serial_generic.py
 pkgs/serialization/serial_union.py
 pkgs/serialization/unknown.py
+pkgs/serialization/usage.py
 pkgs/serialization/yaml.py
 pkgs/serialization_util/__init__.py
 pkgs/serialization_util/_get_type_for_serialization.py
@@ -159,6 +161,8 @@ uncountable/integration/executors/script_executor.py
 uncountable/integration/http_server/__init__.py
 uncountable/integration/http_server/types.py
 uncountable/integration/queue_runner/__init__.py
+uncountable/integration/queue_runner/job_process.py
+uncountable/integration/queue_runner/job_runner.py
 uncountable/integration/queue_runner/job_scheduler.py
 uncountable/integration/queue_runner/queue_runner.py
 uncountable/integration/queue_runner/types.py
@@ -208,6 +212,8 @@ uncountable/types/curves.py
 uncountable/types/curves_t.py
 uncountable/types/data.py
 uncountable/types/data_t.py
+uncountable/types/dataset_measurements.py
+uncountable/types/dataset_measurements_t.py
 uncountable/types/entity.py
 uncountable/types/entity_t.py
 uncountable/types/experiment_groups.py
@@ -420,6 +426,8 @@ uncountable/types/api/recipes/upsert_step_relationships.py
 uncountable/types/api/runsheet/__init__.py
 uncountable/types/api/runsheet/complete_async_upload.py
 uncountable/types/api/runsheet/export_default_runsheet.py
+uncountable/types/api/time_series/__init__.py
+uncountable/types/api/time_series/ingest_dataset_measurements.py
 uncountable/types/api/triggers/__init__.py
 uncountable/types/api/triggers/run_trigger.py
 uncountable/types/api/uploader/__init__.py

{uncountablepythonsdk-0.0.179 → uncountablepythonsdk-0.0.180}/examples/integration-server/jobs/materials_auto/example_http.py

@@ -23,9 +23,9 @@ class HttpExample(CustomHttpJob):
     @staticmethod
     def validate_request(
         *,
-        request: GenericHttpRequest,  # noqa: ARG004
-        job_definition: job_definition_t.HttpJobDefinitionBase,  # noqa: ARG004
-        profile_meta: job_definition_t.ProfileMetadata,  # noqa: ARG004
+        request: GenericHttpRequest,  # ruff:ignore[unused-static-method-argument]
+        job_definition: job_definition_t.HttpJobDefinitionBase,  # ruff:ignore[unused-static-method-argument]
+        profile_meta: job_definition_t.ProfileMetadata,  # ruff:ignore[unused-static-method-argument]
     ) -> None:
         if (
             CustomHttpJob.get_validated_oauth_request_user_id(
@@ -40,8 +40,8 @@ class HttpExample(CustomHttpJob):
     @staticmethod
     def handle_request(
         *,
-        request: GenericHttpRequest,  # noqa: ARG004
-        job_definition: job_definition_t.HttpJobDefinitionBase,  # noqa: ARG004
-        profile_meta: job_definition_t.ProfileMetadata,  # noqa: ARG004
+        request: GenericHttpRequest,  # ruff:ignore[unused-static-method-argument]
+        job_definition: job_definition_t.HttpJobDefinitionBase,  # ruff:ignore[unused-static-method-argument]
+        profile_meta: job_definition_t.ProfileMetadata,  # ruff:ignore[unused-static-method-argument]
     ) -> GenericHttpResponse:
         return GenericHttpResponse(response="OK", status_code=200)

{uncountablepythonsdk-0.0.179 → uncountablepythonsdk-0.0.180}/examples/integration-server/jobs/materials_auto/example_instrument.py

@@ -84,10 +84,9 @@ class InstrumentExample(WebhookJob[InstrumentPayload]):
                         num_users = len(message.user_ids)
                         if num_users <= 1:
                             break
-                        else:
-                            args.logger.log_info(
-                                f"Session still open, {num_users} users in room."
-                            )
+                        args.logger.log_info(
+                            f"Session still open, {num_users} users in room."
+                        )
                     case _:
                         args.logger.log_info("Session still open...")
                         continue

uncountablepythonsdk-0.0.180/examples/integration-server/pyproject.toml

@@ -0,0 +1,223 @@
+[project]
+# begin_project_name
+# DO NOT MODIFY
+name = "integration-server-testing"
+# end_project_name
+dynamic = ["version"]
+dependencies = [
+  "mypy >= 2.1.0, < 3",
+  "ruff == 0.*",
+  "openpyxl == 3.*",
+  "more_itertools ==11.*",
+  "types-paramiko ==5.0.0.20260617",
+  "types-openpyxl == 3.*",
+  "types-pytz ==2026.*",
+  "types-requests == 2.*",
+  "types-simplejson == 3.*",
+  "pandas-stubs",
+  "xlrd == 2.*",
+  "msgspec ==0.21.*",
+  "websockets==16.1.1",
+]
+
+[tool.mypy]
+strict = true
+show_error_codes = true
+check_untyped_defs = true
+disallow_any_generics = false
+disallow_incomplete_defs = true
+disallow_subclassing_any = false
+disallow_untyped_calls = false
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+implicit_reexport = true
+no_implicit_optional = true
+no_implicit_reexport = true
+strict_equality = true
+warn_redundant_casts = true
+warn_unused_ignores = false
+warn_no_return = true
+warn_return_any = true
+local_partial_types = true
+warn_unreachable = false
+warn_unused_configs = true
+exclude = [
+    "env"
+]
+
+[tool.ruff]
+preview = true
+lint.select = ['ALL']
+lint.ignore = [
+    "builtin-variable-shadowing", # Variable shadows python built-in. Nice to have
+    "builtin-argument-shadowing", # Variable shadows python built-in. Nice to have
+    "builtin-attribute-shadowing", # Class attribute shadows python built-in. Nice to have
+    "ANN",  # Missing type annotations. Skip in favor of mypy
+    "ARG",  # Unused arguments. Skip
+    "mutable-argument-default", # Do not use mutable data structures for argument defaults. Priority.
+    "function-uses-loop-variable", # Function doesn't bind loop variable.  Priority
+    "raise-without-from-inside-except", # Exception handling raise from.  Skip
+    "BLE", # Blind exception.  Should Add
+    "unnecessary-generator-set", # Unecessary generator.  Skip
+    "unnecessary-list-comprehension-set", # Unnecessary list comprehension.  Skip
+    "unnecessary-literal-set", # Unnecessary list. rewrite as set.  Skip
+    "unnecessary-collection-call", # Unnecessary dict call.  Skip
+    "unnecessary-comprehension", # Unnecessary list comprehension.  Skip
+    "unnecessary-map", # Unnecessary use of map.  Should Add
+    "missing-trailing-comma", # Trailing comma.  Too opinionated.  Skip
+    "CPY", # Copyright notices.  Skip
+    "D",  # Docstrings.  Skip
+    "DOC", # Docstrings.  Skip
+    "call-datetime-without-tzinfo",  # Missing tzinfo.  Should add
+    "call-datetime-today",  # Use of today instead of now.  Should add
+    "call-datetime-utcnow",  # Use of utcnow not allowed.  Should add
+    "call-datetime-utcfromtimestamp",  # Use of utcfrom timestamp not allowed.  Should add
+    "call-datetime-now-without-tzinfo",  # now without tz not allowed.  Should add
+    "call-datetime-fromtimestamp",  # use of from timestamp without tz not allowed.  Should add
+    "call-datetime-strptime-without-zone",  # replace timezone after strptime.  Should add
+    "call-date-today",  # Use of today not allowed.  Should add
+    "line-too-long",  # Line length.  Skip
+    "raw-string-in-exception", # exception must not use a string literal.  skip
+    "f-string-in-exception", # exception must not use an f string.  skip
+    "dot-format-in-exception", # exception must not use a format string. skip
+    "ERA", # Rules around commented code.  Skip
+    "shebang-missing-executable-file", # Executable files need shebang.  Skip
+    "unused-variable", # TEMPORARY DISABLE TO CLEAR UP BUILD.  Unused variables.  Prioritize.
+    "boolean-type-hint-positional-argument", # boolean positional args.  Skip
+    "boolean-default-value-positional-argument", # boolean default positional.  Skip
+    "boolean-positional-value-in-call", # boolean positional. Skip
+    "line-contains-fixme", # Fixme comment. Skip
+    "line-contains-todo", # todo comment.  Skip
+    "line-contains-hack", # hack comment.  Skip
+    "static-join-to-f-string", # Use F-String instead of join.  Should Add
+    "read-whole-file", # replace read function. Skip
+    "write-whole-file", # opinionated file opening. Skip
+    "repeated-append", # extend vs append.  Skip
+    "reimplemented-operator", # opinionated rules about lambdas. skip
+    "reimplemented-starmap", # starmap vs generator.  Skip
+    "logging-f-string",  # avoid f strings in log statements.  Should add
+    "implicit-namespace-package", # implicit imports.  use __init__.  Skip
+    "single-line-implicit-string-concatenation", # rule conflicts with formatter. Skip
+    "explicit-string-concatenation", # Explicitly concatenated string should be implicitly concatenated. Skip
+    "invalid-class-name", # class name case. Skip
+    "invalid-function-name", # function name lower case. Skip
+    "invalid-argument-name", # argument name lower cas. Skip
+    "invalid-first-argument-name-for-class-method", # cls_ first variable.  Should add
+    "non-lowercase-variable-in-function", # lower case variable.  Skip
+    "lowercase-imported-as-non-lowercase", # case consistency on import.  Skip
+    "mixed-case-variable-in-class-scope", # mix casing, skip
+    "camelcase-imported-as-acronym", # don't use acronym.  Should add
+    "error-suffix-on-exception-name", # exception override should be named with error. should add
+    "invalid-module-name", # module naming.  Should add
+    "pandas-use-of-inplace-argument", # avoid inplace.  Skip
+    "pandas-use-of-dot-not-null", # .notna.  Should add
+    "pandas-use-of-dot-pivot-or-unstack", # .pivottable.  Should add
+    "pandas-use-of-dot-values", # use .to_numpy.  Skip
+    "pandas-use-of-pd-merge", # use .merge.  Should add
+    "try-except-in-loop", # avoid try except in loop. Skip
+    "manual-list-comprehension", # use list comprehension. Skip
+    "manual-list-copy", # use list.copy.  Skip
+    "manual-dict-comprehension", # use dictionary comprehension.  Skip
+    "suspicious-eval-usage", # avoid eval.  Should add
+    "blanket-type-ignore", # ignore specific rules.  Skip
+    "blanket-noqa", # ignore specific rules.  Skip
+    "useless-import-alias", # import alias doesn't rename. Skip
+    "import-outside-top-level", # import at top level.  Skip
+    "compare-to-empty-string", # falsy empty string.  Skip
+    "comparison-with-itself", # self comparison.  Skip
+    "manual-from-import", # alias import.  Skip
+    "too-many-return-statements", # too many return statements.  Skip
+    "too-many-branches", # too many branches.  Skip
+    "too-many-arguments", # too many arguments.  Skip
+    "too-many-statements", # too many statements.  Skip
+    "too-many-boolean-expressions", # too many expressions. Skip
+    "too-many-positional-arguments", # too many positional arguments.  Skip
+    "repeated-equality-comparison", # multiple comparisons.  Skip
+    "magic-value-comparison", # magic values.  skip
+    "collapsible-else-if", # use elif instead of else if.  skip
+    "non-augmented-assignment", # use augmented assignment.  confuses strings, wait for fix.  skip
+    "literal-membership", # use a set literal for membership test.  skip
+    "no-self-use", # method inference.  skip
+    "unnecessary-lambda", # inline lambda.  skip
+    "useless-else-on-loop", # else without break. skip
+    "global-variable-not-assigned", # global withotu assignment.  consider
+    "global-statement", # discourage global.  skip
+    "subprocess-run-without-check", # subprocess run without check arg.  skip
+    "unspecified-encoding", # open without encoding. skip
+    "redefined-loop-name", # for loop variable overwritten.  skip
+    "pytest-fixture-incorrect-parentheses-style", # change ficture decorator.  should add
+    "pytest-raises-too-broad", # use more specific raises decorator.  should add
+    "pytest-composite-assertion", # break down complex assertsions.  skip
+    "PTH", # enforces new python path library usage.  skip
+    "unnecessary-literal-union", # turn union of literals to just literal with multiple. should add
+    "redundant-numeric-union", # use float instead of int | float. should remove floats
+    "RET", # enforces return semantics. break up and re-evaluate later
+    "unnecessary-paren-on-raise-exception", # unnecessary parens on exception.  skip.
+    "ambiguous-unicode-character-string", # ambiguous characters.  skip
+    "collection-literal-concatenation", # prefer star to concatenation.  skip
+    "explicit-f-string-type-conversion", # use explicit conversion flag.  skip
+    "mutable-class-default", # type with class var.  should add
+    "unnecessary-iterable-allocation-for-first-element", # prefer next for single element. skip
+    "unused-noqa", # unused noqa.  should add
+    "S", # should split up
+    "assert", # allow asserts. skip
+    "collapsible-if", # use single if instead of nested. skip
+    "suppressible-exception", # use exception supress. skip
+    "if-else-block-instead-of-if-exp", # use ternary.  skip
+    "reimplemented-builtin", # use any/all.  should add
+    "uncapitalized-environment-variables", # case of env variable.  skip
+    "if-with-same-arms", # combine if branches.  skip
+    "open-file-with-context-handler", # use context handler for files.  should add
+    "multiple-with-statements", # use single with.  should add
+    "private-member-access", # private member access.  should add
+    "no-slots-in-str-subclass", # subclass of string should define slot.  should add
+    "print", # print.  skip
+    "p-print", # pprint.  skip
+    "typing-only-first-party-import", # move application import into type checking.  skip
+    "typing-only-third-party-import", # move third party import into type checking.  consider
+    "typing-only-standard-library-import", # move library into type checking lbock.  consider
+    "TD",  # todo comments.  skip
+    "relative-imports", # eliminate relative imports from parents.  skip
+    "raise-vanilla-class", # create your own exception. skip
+    "raise-vanilla-args", # avoid long messages.  skip
+    "type-check-without-type-error", # prefer type error.  should add
+    "verbose-raise", # raise without exception name. consider
+    "try-consider-else", # move statement to else block.  consider
+    "raise-within-try", # move raise to inner function. skip
+    "non-pep604-annotation-union", # use union for type annotations.  should add
+    "convert-typed-dict-functional-to-class", # eliminate typeddict.  should add
+    "printf-string-formatting", # use format specifiers.  should add
+    "zip-without-explicit-strict", # zip strict. we have a deprecated pattern here, consider
+    "outdated-version-block", # outdated version block, consider
+    "zip-instead-of-pairwise", # prefer itertools.pairwise over zip, consider
+    "unsorted-dunder-all",  # __all__ is not sorted.  skip due to isort complication
+    "datetime-timezone-utc", # use datetime.UTC, TODO add back in
+    "deprecated-import", # replacing List with list, TODO add back in
+    # ## FROM RUFF UPGRADE
+    "import-private-name",  # private name imports.  should add
+    "too-many-nested-blocks",  # too many nested blocks -- add with config.  skip
+    "unnecessary-dict-comprehension-for-iterable",  # unnecessary dict comprehension.  skip
+    "enumerate-for-loop",  # index variable should use enumerate.  should add.
+    "too-many-locals"  # too many local variables.  configure below.  should add with config
+]
+line-length = 90
+target-version = "py311"
+
+exclude = [
+    'env'
+]
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["unused-import"]
+
+[tool.ruff.lint.isort]
+split-on-trailing-comma = true
+known-first-party = ["pkgs"]
+
+[tool.ruff.lint.mccabe]
+max-complexity = 130 # goal would be to bring this down to ~50 or so
+
+[tool.ruff.lint.pylint]
+max-locals=50
+
+[tool.setuptools]
+py-modules = []

uncountablepythonsdk-0.0.180/examples/stream_dataset_measurements.py

@@ -0,0 +1,196 @@
+import argparse
+import datetime
+import json
+import os
+import random
+import time
+from decimal import Decimal
+
+import requests
+from uncountable.types import dataset_measurements_t, identifier_t
+from uncountable.types.api.time_series import ingest_dataset_measurements as api
+
+from pkgs.serialization_util import serialize_for_api
+
+API_ID_ENV_VAR = "UNC_API_ID"
+API_SECRET_KEY_ENV_VAR = "UNC_API_SECRET_KEY"
+MEASUREMENTS_PER_SECOND = 500
+ONE_SECOND = datetime.timedelta(seconds=1)
+VALUE_QUANTUM = Decimal("0.0001")
+
+
+def _random_value(*, min_value: Decimal, max_value: Decimal) -> Decimal:
+    span = max_value - min_value
+    sampled = (min_value + span * Decimal(str(random.random()))).quantize(VALUE_QUANTUM)
+    return max(min_value, min(max_value, sampled))
+
+
+def _window_sizes(*, num_measurements: int) -> list[int]:
+    full_windows, remainder = divmod(num_measurements, MEASUREMENTS_PER_SECOND)
+    window_sizes = [MEASUREMENTS_PER_SECOND] * full_windows
+    if remainder:
+        window_sizes.append(remainder)
+    return window_sizes
+
+
+def _build_measurements(
+    *,
+    anchor: datetime.datetime,
+    window_index: int,
+    measurement_count: int,
+    min_value: Decimal,
+    max_value: Decimal,
+) -> list[dataset_measurements_t.Measurement]:
+    window_start = anchor + window_index * ONE_SECOND
+    step = ONE_SECOND / measurement_count
+    return [
+        dataset_measurements_t.Measurement(
+            observation_timestamp=window_start + index * step,
+            value=dataset_measurements_t.NumericMeasurementValue(
+                value=_random_value(min_value=min_value, max_value=max_value)
+            ),
+        )
+        for index in range(measurement_count)
+    ]
+
+
+def _sleep_until(instant: datetime.datetime) -> None:
+    remaining_seconds = (instant - datetime.datetime.now(datetime.UTC)).total_seconds()
+    if remaining_seconds > 0:
+        time.sleep(remaining_seconds)
+
+
+def _post_measurements(
+    *,
+    session: requests.Session,
+    url: str,
+    dataset_id: int,
+    measurements: list[dataset_measurements_t.Measurement],
+) -> None:
+    arguments = api.Arguments(
+        dataset_key=identifier_t.IdentifierKeyId(id=dataset_id),
+        measurements=measurements,
+    )
+    body = {"data": json.dumps(serialize_for_api(arguments))}
+    response = session.post(url, data=body)
+    response.raise_for_status()
+
+
+def _stream(
+    *,
+    session: requests.Session,
+    url: str,
+    dataset_id: int,
+    num_measurements: int,
+    min_value: Decimal,
+    max_value: Decimal,
+) -> None:
+    anchor = datetime.datetime.now(datetime.UTC)
+    window_sizes = _window_sizes(num_measurements=num_measurements)
+    print(
+        f"streaming {num_measurements} measurement(s) to dataset {dataset_id} "
+        f"at {MEASUREMENTS_PER_SECOND}/second across {len(window_sizes)} second(s)",
+        flush=True,
+    )
+    streamed = 0
+    for window_index, measurement_count in enumerate(window_sizes):
+        measurements = _build_measurements(
+            anchor=anchor,
+            window_index=window_index,
+            measurement_count=measurement_count,
+            min_value=min_value,
+            max_value=max_value,
+        )
+        _sleep_until(anchor + (window_index + 1) * ONE_SECOND)
+        _post_measurements(
+            session=session,
+            url=url,
+            dataset_id=dataset_id,
+            measurements=measurements,
+        )
+        streamed += measurement_count
+        percent_complete = 100 * streamed / num_measurements
+        print(
+            f"streamed {streamed}/{num_measurements} measurement(s) "
+            f"({percent_complete:.1f}%)",
+            flush=True,
+        )
+    print(f"done: streamed {streamed} measurement(s)", flush=True)
+
+
+def _parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(
+        description=(
+            "Stream random measurements to a monitor (timeseries) dataset via the "
+            "dataset stream API."
+        ),
+        allow_abbrev=False,
+    )
+    parser.add_argument(
+        "--base-url",
+        required=True,
+        help="Base URL of the Uncountable server, e.g. http://localhost:5000",
+    )
+    parser.add_argument(
+        "--dataset-id",
+        type=int,
+        required=True,
+        help="Object id of the monitor dataset to receive measurements",
+    )
+    parser.add_argument(
+        "--num-measurements",
+        type=int,
+        required=True,
+        help="Total number of measurements to stream",
+    )
+    parser.add_argument(
+        "--min-value",
+        type=Decimal,
+        required=True,
+        help="Minimum random measurement value",
+    )
+    parser.add_argument(
+        "--max-value",
+        type=Decimal,
+        required=True,
+        help="Maximum random measurement value",
+    )
+    args = parser.parse_args()
+    if args.num_measurements <= 0:
+        parser.error("--num-measurements must be positive")
+    if args.max_value < args.min_value:
+        parser.error("--max-value must be greater than or equal to --min-value")
+    return args
+
+
+def main() -> None:
+    args = _parse_args()
+
+    api_id = os.environ.get(API_ID_ENV_VAR)
+    api_secret_key = os.environ.get(API_SECRET_KEY_ENV_VAR)
+    if not api_id or not api_secret_key:
+        missing = [
+            name
+            for name, value in (
+                (API_ID_ENV_VAR, api_id),
+                (API_SECRET_KEY_ENV_VAR, api_secret_key),
+            )
+            if not value
+        ]
+        raise SystemExit(
+            f"missing required environment variable(s): {', '.join(missing)}"
+        )
+
+    session = requests.Session()
+    session.auth = (api_id, api_secret_key)
+    _stream(
+        session=session,
+        url=f"{args.base_url.rstrip('/')}/{api.ENDPOINT_PATH}",
+        dataset_id=args.dataset_id,
+        num_measurements=args.num_measurements,
+        min_value=args.min_value,
+        max_value=args.max_value,
+    )
+
+
+main()

{uncountablepythonsdk-0.0.179 → uncountablepythonsdk-0.0.180}/pkgs/argument_parser/__init__.py

@@ -2,6 +2,7 @@
 
 __all__ = [
     "CachedParser",
+    "InternalCachedParser",
     "ParserBase",
     "ParserEnumError",
     "ParserError",
@@ -10,6 +11,7 @@ __all__ = [
     "ParserOptions",
     "ParserTypeError",
     "SourceEncoding",
+    "TestCachedParser",
     "build_parser",
     "camel_to_snake_case",
     "is_missing",
@@ -20,7 +22,11 @@ __all__ = [
     "snake_to_camel_case",
 ]
 
-from .cached_parser import CachedParser as CachedParser
+from .cached_parser import (
+    CachedParser as CachedParser,
+    InternalCachedParser as InternalCachedParser,
+    TestCachedParser as TestCachedParser,
+)
 from .case_convert import (
     camel_to_snake_case as camel_to_snake_case,
     kebab_to_pascal_case as kebab_to_pascal_case,