pytest-pickle-cache 0.1.0__tar.gz

pytest_pickle_cache-0.1.0/.devcontainer/devcontainer.json

@@ -0,0 +1,18 @@
+{
+  "image": "mcr.microsoft.com/vscode/devcontainers/base:ubuntu24.04",
+  "features": {
+    "ghcr.io/devcontainers-contrib/features/starship:1": {}
+  },
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "charliermarsh.ruff",
+        "fill-labs.dependi",
+        "ms-python.python",
+        "ms-python.vscode-pylance",
+        "tamasfe.even-better-toml"
+      ]
+    }
+  },
+  "postCreateCommand": ".devcontainer/postCreate.sh"
+}

pytest_pickle_cache-0.1.0/.devcontainer/postCreate.sh

@@ -0,0 +1,10 @@
+#!/bin/bash
+
+echo 'eval "$(starship init bash)"' >> ~/.bashrc
+mkdir -p ~/.config
+cp .devcontainer/starship.toml ~/.config
+
+curl -LsSf https://astral.sh/uv/install.sh | sh
+source $HOME/.cargo/env
+uv tool install ruff@latest
+echo 'eval "$(uv generate-shell-completion bash)"' >> ~/.bashrc

pytest_pickle_cache-0.1.0/.devcontainer/starship.toml

@@ -0,0 +1,29 @@
+"$schema" = 'https://starship.rs/config-schema.json'
+
+add_newline = true
+
+[username]
+disabled = true
+
+[hostname]
+disabled = true
+
+[package]
+format = '[$symbol$version]($style) '
+symbol = "󰏗 "
+style = 'bold blue'
+
+[git_branch]
+format = '[$symbol$branch(:$remote_branch)]($style)'
+symbol = ' '
+style = 'bold green'
+
+[git_status]
+style = 'red'
+modified = '*'
+format = '([$modified]($style)) '
+
+[python]
+format = '[${symbol}${pyenv_prefix}(${version} )(\($virtualenv\) )]($style)'
+symbol = ' '
+style = 'yellow'

pytest_pickle_cache-0.1.0/.gitignore

@@ -0,0 +1,6 @@
+.coverage
+.venv/
+__pycache__/
+dist/
+lcov.info
+uv.lock

pytest_pickle_cache-0.1.0/LICENSE

@@ -0,0 +1,9 @@
+MIT License
+
+Copyright (c) 2020-present daizutabi <daizutabi@gmail.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

pytest_pickle_cache-0.1.0/PKG-INFO

@@ -0,0 +1,154 @@
+Metadata-Version: 2.4
+Name: pytest-pickle-cache
+Version: 0.1.0
+Summary: A pytest plugin for caching test results using pickle.
+Project-URL: Documentation, https://daizutabi.github.io/pytest-pickle-cache/
+Project-URL: Source, https://github.com/daizutabi/pytest-pickle-cache
+Project-URL: Issues, https://github.com/daizutabi/pytest-pickle-cache/issues
+Author-email: daizutabi <daizutabi@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2020-present daizutabi <daizutabi@gmail.com>
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+License-File: LICENSE
+Classifier: Framework :: Pytest
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Testing
+Requires-Python: >=3.10
+Requires-Dist: pytest>=7
+Description-Content-Type: text/markdown
+
+# pytest-pickle-cache
+
+[![PyPI Version][pypi-v-image]][pypi-v-link]
+[![Python Version][python-v-image]][python-v-link]
+[![Build Status][GHAction-image]][GHAction-link]
+[![Coverage Status][codecov-image]][codecov-link]
+
+## Overview
+
+`pytest-pickle-cache` is a pytest plugin for caching test results using pickle.
+By utilizing this plugin, you can reduce test execution time and perform tests
+more efficiently.
+
+## Installation
+
+You can install `pytest-pickle-cache` using the following command:
+
+```bash
+pip install pytest-pickle-cache
+```
+
+## Fixture
+
+The `use_cache` fixture is a pytest fixture that provides a caching mechanism
+for pytest, allowing you to store and retrieve objects using a specified key.
+The objects are serialized and deserialized using pickle and base64 encoding.
+
+```python
+def use_cache(key: str, func: Callable[[], Any]) -> Any:
+    """Retrieve a cached result or execute the function if not cached.
+
+    Args:
+        key (str): The key to identify the cached result.
+        func (Callable[[], Any]): The function to execute if the result is
+            not cached. The result of the function is serialized and stored
+            in the cache for future use.
+
+    Returns:
+        Any: The cached result or the result of the executed function.
+    """
+```
+
+## Example
+
+Here is a specific example of how to use `pytest-pickle-cache` to cache test results.
+
+```python
+import datetime
+
+import pytest
+from pandas import DataFrame
+
+
+def create() -> DataFrame:
+    """Create a DataFrame with the current time."""
+    now = datetime.datetime.now()
+    return DataFrame({"now": [now]})
+
+
+def test_create(use_cache):
+    """Create a DataFrame using cache and compare the results."""
+    # Retrieve DataFrame using cache
+    df_cached = use_cache("key", create)
+
+    # Create a new DataFrame
+    df_created = create()
+
+    # Assert that the cached DataFrame and the newly created DataFrame are different.
+    assert not df_created.equals(df_cached)
+
+
+def test_create_with_cache(use_cache):
+    """Use cache to retrieve the same DataFrame and ensure the results are the same."""
+    # Cache the DataFrame on the first call
+    df_cached_first = use_cache("key", create)
+
+    # Call the same function again to retrieve from cache
+    df_cached_second = use_cache("key", create)
+
+    # Assert that the cached DataFrame is the same on the second call.
+    assert df_cached_first.equals(df_cached_second)
+```
+
+
+You can also use `use_cache` fixture as a fixture in your test file.
+
+```python
+@pytest.fixture
+def df(use_cache):
+    return use_cache("key", create)
+```
+
+You can also use `use_cache` fixture with a parametrized fixture.
+
+```python
+def create(param: int) -> DataFrame:
+    """Create a DataFrame with the current time."""
+    now = datetime.datetime.now()
+    return DataFrame({"now": [now], "param": [param]})
+
+
+@pytest.fixture(params=[1, 2, 3])
+def df(use_cache, request):
+    return use_cache(f"key_{request.param}", lambda: create(request.param))
+```
+
+## Benefits of this Example
+
+- **Efficiency in Testing**: By using `pytest-pickle-cache`, you can avoid running
+the same test multiple times, reducing the overall test execution time.
+
+- **Consistency of Results**: Using cache ensures that you get the same result
+for the same input, maintaining consistency in your tests.
+
+<!-- Badges -->
+[pypi-v-image]: https://img.shields.io/pypi/v/pytest-pickle-cache.svg
+[pypi-v-link]: https://pypi.org/project/pytest-pickle-cache/
+[python-v-image]: https://img.shields.io/pypi/pyversions/pytest-pickle-cache.svg
+[python-v-link]: https://pypi.org/project/pytest-pickle-cache
+[GHAction-image]: https://github.com/daizutabi/pytest-pickle-cache/actions/workflows/ci.yml/badge.svg?branch=main&event=push
+[GHAction-link]: https://github.com/daizutabi/pytest-pickle-cache/actions?query=event%3Apush+branch%3Amain
+[codecov-image]: https://codecov.io/github/daizutabi/pytest-pickle-cache/coverage.svg?branch=main
+[codecov-link]: https://codecov.io/github/daizutabi/pytest-pickle-cache?branch=main

pytest_pickle_cache-0.1.0/README.md

@@ -0,0 +1,123 @@
+# pytest-pickle-cache
+
+[![PyPI Version][pypi-v-image]][pypi-v-link]
+[![Python Version][python-v-image]][python-v-link]
+[![Build Status][GHAction-image]][GHAction-link]
+[![Coverage Status][codecov-image]][codecov-link]
+
+## Overview
+
+`pytest-pickle-cache` is a pytest plugin for caching test results using pickle.
+By utilizing this plugin, you can reduce test execution time and perform tests
+more efficiently.
+
+## Installation
+
+You can install `pytest-pickle-cache` using the following command:
+
+```bash
+pip install pytest-pickle-cache
+```
+
+## Fixture
+
+The `use_cache` fixture is a pytest fixture that provides a caching mechanism
+for pytest, allowing you to store and retrieve objects using a specified key.
+The objects are serialized and deserialized using pickle and base64 encoding.
+
+```python
+def use_cache(key: str, func: Callable[[], Any]) -> Any:
+    """Retrieve a cached result or execute the function if not cached.
+
+    Args:
+        key (str): The key to identify the cached result.
+        func (Callable[[], Any]): The function to execute if the result is
+            not cached. The result of the function is serialized and stored
+            in the cache for future use.
+
+    Returns:
+        Any: The cached result or the result of the executed function.
+    """
+```
+
+## Example
+
+Here is a specific example of how to use `pytest-pickle-cache` to cache test results.
+
+```python
+import datetime
+
+import pytest
+from pandas import DataFrame
+
+
+def create() -> DataFrame:
+    """Create a DataFrame with the current time."""
+    now = datetime.datetime.now()
+    return DataFrame({"now": [now]})
+
+
+def test_create(use_cache):
+    """Create a DataFrame using cache and compare the results."""
+    # Retrieve DataFrame using cache
+    df_cached = use_cache("key", create)
+
+    # Create a new DataFrame
+    df_created = create()
+
+    # Assert that the cached DataFrame and the newly created DataFrame are different.
+    assert not df_created.equals(df_cached)
+
+
+def test_create_with_cache(use_cache):
+    """Use cache to retrieve the same DataFrame and ensure the results are the same."""
+    # Cache the DataFrame on the first call
+    df_cached_first = use_cache("key", create)
+
+    # Call the same function again to retrieve from cache
+    df_cached_second = use_cache("key", create)
+
+    # Assert that the cached DataFrame is the same on the second call.
+    assert df_cached_first.equals(df_cached_second)
+```
+
+
+You can also use `use_cache` fixture as a fixture in your test file.
+
+```python
+@pytest.fixture
+def df(use_cache):
+    return use_cache("key", create)
+```
+
+You can also use `use_cache` fixture with a parametrized fixture.
+
+```python
+def create(param: int) -> DataFrame:
+    """Create a DataFrame with the current time."""
+    now = datetime.datetime.now()
+    return DataFrame({"now": [now], "param": [param]})
+
+
+@pytest.fixture(params=[1, 2, 3])
+def df(use_cache, request):
+    return use_cache(f"key_{request.param}", lambda: create(request.param))
+```
+
+## Benefits of this Example
+
+- **Efficiency in Testing**: By using `pytest-pickle-cache`, you can avoid running
+the same test multiple times, reducing the overall test execution time.
+
+- **Consistency of Results**: Using cache ensures that you get the same result
+for the same input, maintaining consistency in your tests.
+
+<!-- Badges -->
+[pypi-v-image]: https://img.shields.io/pypi/v/pytest-pickle-cache.svg
+[pypi-v-link]: https://pypi.org/project/pytest-pickle-cache/
+[python-v-image]: https://img.shields.io/pypi/pyversions/pytest-pickle-cache.svg
+[python-v-link]: https://pypi.org/project/pytest-pickle-cache
+[GHAction-image]: https://github.com/daizutabi/pytest-pickle-cache/actions/workflows/ci.yml/badge.svg?branch=main&event=push
+[GHAction-link]: https://github.com/daizutabi/pytest-pickle-cache/actions?query=event%3Apush+branch%3Amain
+[codecov-image]: https://codecov.io/github/daizutabi/pytest-pickle-cache/coverage.svg?branch=main
+[codecov-link]: https://codecov.io/github/daizutabi/pytest-pickle-cache?branch=main

pytest_pickle_cache-0.1.0/pyproject.toml

@@ -0,0 +1,64 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "pytest-pickle-cache"
+version = "0.1.0"
+authors = [{ name = "daizutabi", email = "daizutabi@gmail.com" }]
+description = "A pytest plugin for caching test results using pickle."
+readme = "README.md"
+license = { file = "LICENSE" }
+classifiers = [
+    "Framework :: Pytest",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Topic :: Software Development :: Testing",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+]
+requires-python = ">=3.10"
+dependencies = ["pytest>=7"]
+
+[project.urls]
+Documentation = "https://daizutabi.github.io/pytest-pickle-cache/"
+Source = "https://github.com/daizutabi/pytest-pickle-cache"
+Issues = "https://github.com/daizutabi/pytest-pickle-cache/issues"
+
+[project.entry-points.pytest11]
+pickle_cache = "pytest_pickle_cache.plugin"
+
+[tool.uv]
+dev-dependencies = ["pandas>=2", "pytest-cov>=6.0"]
+
+[tool.hatch.build.targets.sdist]
+exclude = ["/.github", "/docs"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/pytest_pickle_cache"]
+
+[tool.pytest.ini_options]
+addopts = [
+    "--doctest-modules",
+    "--cov=pytest_pickle_cache",
+    "--cov-report=lcov:lcov.info",
+]
+doctest_optionflags = ["NORMALIZE_WHITESPACE", "IGNORE_EXCEPTION_DETAIL"]
+filterwarnings = []
+
+[tool.coverage.run]
+disable_warnings = ["module-not-measured"]
+
+[tool.coverage.report]
+exclude_lines = ["no cov", "raise NotImplementedError", "if TYPE_CHECKING:"]
+skip_covered = true
+
+[tool.ruff]
+line-length = 88
+target-version = "py311"
+
+[tool.ruff.lint]
+unfixable = ["F401"]

pytest_pickle_cache-0.1.0/src/pytest_pickle_cache/plugin.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+import base64
+import binascii
+import pickle
+from typing import TYPE_CHECKING
+
+import pytest
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+    from typing import Any
+
+
+@pytest.fixture(scope="session")
+def use_cache(request: pytest.FixtureRequest) -> Callable[[str, Callable], Any]:
+    """A pytest fixture to use cache.
+
+    This fixture provides a caching mechanism for pytest, allowing you to
+    store and retrieve objects using a specified key. The objects are serialized
+    and deserialized using pickle and base64 encoding.
+    """
+
+    def use_cache(key: str, create: Callable[[], object]) -> Any:
+        try:
+            if value := request.config.cache.get(key, None):
+                return pickle.loads(base64.b64decode(value))
+        except (pickle.UnpicklingError, binascii.Error):
+            request.config.cache.set(key, None)
+
+        obj = create()
+        value = base64.b64encode(pickle.dumps(obj)).decode("utf-8")
+
+        request.config.cache.set(key, value)
+
+        return obj
+
+    return use_cache

pytest_pickle_cache-0.1.0/tests/test_plugin.py

@@ -0,0 +1,30 @@
+"""Test the pytest_pickle_cache plugin."""
+
+import datetime
+import time
+
+import pytest
+from pandas import DataFrame
+
+
+def create() -> DataFrame:
+    """Create a DataFrame with the current time."""
+    now = datetime.datetime.now()
+    return DataFrame({"now": [now]})
+
+
+@pytest.fixture
+def df(use_cache):
+    return use_cache("use_cache", create)
+
+
+def test_create(use_cache):
+    df_cached = use_cache("use_cache", create)
+    time.sleep(1)
+    df_created = create()
+    assert not df_created.equals(df_cached)
+
+
+def test_create_df(df: DataFrame, use_cache):
+    df_cached = use_cache("use_cache", create)
+    assert df.equals(df_cached)