38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
# This file is part of Hypothesis, which may be found at
|
|
# https://github.com/HypothesisWorks/hypothesis/
|
|
#
|
|
# Copyright the Hypothesis Authors.
|
|
# Individual contributors are listed in AUTHORS.rst and the git log.
|
|
#
|
|
# This Source Code Form is subject to the terms of the Mozilla Public License,
|
|
# v. 2.0. If a copy of the MPL was not distributed with this file, You can
|
|
# obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
"""Run all functions registered for the "hypothesis" entry point.
|
|
|
|
This can be used with `st.register_type_strategy` to register strategies for your
|
|
custom types, running the relevant code when *hypothesis* is imported instead of
|
|
your package.
|
|
"""
|
|
|
|
import importlib.metadata
|
|
import os
|
|
|
|
|
|
def get_entry_points():
|
|
try:
|
|
eps = importlib.metadata.entry_points(group="hypothesis")
|
|
except TypeError: # pragma: no cover
|
|
# Load-time selection requires Python >= 3.10. See also
|
|
# https://importlib-metadata.readthedocs.io/en/latest/using.html
|
|
eps = importlib.metadata.entry_points().get("hypothesis", [])
|
|
yield from eps
|
|
|
|
|
|
def run():
|
|
if not os.environ.get("HYPOTHESIS_NO_PLUGINS"):
|
|
for entry in get_entry_points(): # pragma: no cover
|
|
hook = entry.load()
|
|
if callable(hook):
|
|
hook()
|