# Copyright (c) 2013 Potential Ventures Ltd
# Copyright (c) 2013 SolarFlare Communications Inc
# All rights reserved.
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of Potential Ventures Ltd,
# SolarFlare Communications Inc nor the
# names of its contributors may be used to endorse or promote products
# derived from this software without specific prior written permission.
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL POTENTIAL VENTURES LTD BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import ast
import logging as py_logging
import os
import random
import sys
import time
import warnings
from collections.abc import Coroutine
from types import SimpleNamespace
from typing import Any, Dict, List, Union, cast
import cocotb.handle
import cocotb.triggers
from cocotb._scheduler import Scheduler
from cocotb.logging import default_config
from cocotb.regression import RegressionManager, RegressionMode
from cocotb.task import Task
from ._version import __version__
# Things we want in the cocotb namespace
from cocotb.decorators import ( # isort: skip # noqa: F401
external,
function,
test,
parameterize,
)
from cocotb.logging import _filter_from_c, _log_from_c # isort: skip # noqa: F401
log: py_logging.Logger
"""The default cocotb logger."""
_scheduler: Scheduler
"""The global scheduler instance."""
regression_manager: RegressionManager
"""The global regression manager instance."""
argv: List[str]
"""The argument list as seen by the simulator."""
plusargs: Dict[str, Union[bool, str]]
"""A dictionary of "plusargs" handed to the simulation.
See :make:var:`PLUSARGS` for details.
"""
packages: SimpleNamespace
"""A :class:`python:types.SimpleNamespace` of package handles.
This will be populated with handles at test time if packages can be discovered
via the GPI.
.. versionadded:: 2.0
"""
SIM_NAME: str
"""The product information of the running simulator."""
SIM_VERSION: str
"""The version of the running simulator."""
_random_seed: int
"""
The value passed to the Python default random number generator.
See :envvar:`RANDOM_SEED` for details on how the value is computed.
This is guaranteed to hold a value at test time.
"""
_library_coverage: Any = None
""" used for cocotb library coverage """
_user_coverage: Any = None
""" used for user code coverage """
top: cocotb.handle.SimHandleBase
r"""
A handle to the :envvar:`TOPLEVEL` entity/module.
This is equivalent to the :term:`DUT` parameter given to cocotb tests, so it can be used wherever that variable can be used.
It is particularly useful for extracting information about the :term:`DUT` in module-level class and function definitions;
and in parameters to :class:`.TestFactory`\ s.
"""
is_simulation: bool = False
"""``True`` if cocotb was loaded in a simulation."""
def _setup_logging() -> None:
default_config()
global log
log = py_logging.getLogger(__name__)
[docs]
def start_soon(coro: Union[Task, Coroutine]) -> Task:
"""
Schedule a coroutine to be run concurrently.
Note that this is not an ``async`` function,
and the new task will not execute until the calling task yields control.
Args:
coro: A task or coroutine to be run.
Returns:
The :class:`~cocotb.task.Task` that is scheduled to be run.
.. versionadded:: 1.6.0
"""
return _scheduler.start_soon(coro)
[docs]
async def start(coro: Union[Task, Coroutine]) -> Task:
"""
Schedule a coroutine to be run concurrently, then yield control to allow pending tasks to execute.
The calling task will resume execution before control is returned to the simulator.
When the calling task resumes, the newly scheduled task may have completed,
raised an Exception, or be pending on a :class:`~cocotb.triggers.Trigger`.
Args:
coro: A task or coroutine to be run.
Returns:
The :class:`~cocotb.task.Task` that has been scheduled and allowed to execute.
.. versionadded:: 1.6.0
"""
task = _scheduler.start_soon(coro)
await cocotb.triggers.NullTrigger()
return task
[docs]
def create_task(coro: Union[Task, Coroutine]) -> Task:
"""
Construct a coroutine into a :class:`~cocotb.task.Task` without scheduling the task.
The task can later be scheduled with :func:`cocotb.start` or :func:`cocotb.start_soon`.
Args:
coro: An existing task or a coroutine to be wrapped.
Returns:
Either the provided :class:`~cocotb.task.Task` or a new Task wrapping the coroutine.
.. versionadded:: 1.6.0
"""
return cocotb._scheduler.create_task(coro)
def _initialise_testbench(argv_): # pragma: no cover
try:
_start_library_coverage()
_initialise_testbench_(argv_)
except BaseException:
log.exception("cocotb testbench initialization failed. Exiting.")
from cocotb import simulator
simulator.stop_simulator()
_stop_library_coverage()
def _initialise_testbench_(argv_):
# The body of this function is split in two because no coverage is collected on
# the function that starts the coverage. By splitting it in two we get coverage
# on most of the function.
global is_simulation
is_simulation = True
global argv
argv = argv_
# sys.path normally includes "" (the current directory), but does not appear to when python is embedded.
# Add it back because users expect to be able to import files in their test directory.
# TODO: move this to gpi_embed.cpp
sys.path.insert(0, "")
_setup_logging()
# From https://www.python.org/dev/peps/pep-0565/#recommended-filter-settings-for-test-runners
# If the user doesn't want to see these, they can always change the global
# warning settings in their test module.
if not sys.warnoptions:
warnings.simplefilter("default")
from cocotb import simulator
global SIM_NAME, SIM_VERSION
SIM_NAME = simulator.get_simulator_product().strip()
SIM_VERSION = simulator.get_simulator_version().strip()
cocotb.log.info(f"Running on {SIM_NAME} version {SIM_VERSION}")
log.info(
f"Running tests with cocotb v{__version__} from {os.path.dirname(__file__)}"
)
_process_plusargs()
_process_packages()
_setup_random_seed()
_setup_root_handle()
_start_user_coverage()
_setup_regression_manager()
# setup global scheduler system
global _scheduler
_scheduler = Scheduler(test_complete_cb=regression_manager._test_complete)
# start Regression Manager
regression_manager.start_regression()
def _start_library_coverage() -> None: # pragma: no cover
if "COCOTB_LIBRARY_COVERAGE" in os.environ:
try:
import coverage
except ImportError:
log.error(
"cocotb library coverage collection requested but coverage package not available. Install it using `pip install coverage`."
)
else:
global _library_coverage
_library_coverage = coverage.coverage(
data_file=".coverage.cocotb",
config_file=False,
branch=True,
include=[f"{os.path.dirname(__file__)}/*"],
)
_library_coverage.start()
def _stop_library_coverage() -> None:
if _library_coverage is not None:
# TODO: move this once we have normal shutdown behavior to _sim_event
_library_coverage.stop()
_library_coverage.save() # pragma: no cover
def _sim_event(message: str) -> None:
"""Function that can be called externally to signal an event."""
from cocotb.result import SimFailure
# We simply return here as the simulator will exit
# so no cleanup is needed
msg = f"Failing test at simulator request before test run completion: {message}"
if _scheduler is not None:
_scheduler.log.error(msg)
_scheduler._finish_scheduler(SimFailure(msg))
else:
log.error(msg)
_stop_user_coverage()
_stop_library_coverage()
def _process_plusargs() -> None:
global plusargs
plusargs = {}
for option in cocotb.argv:
if option.startswith("+"):
if option.find("=") != -1:
(name, value) = option[1:].split("=", 1)
plusargs[name] = value
else:
plusargs[option[1:]] = True
def _process_packages() -> None:
global packages
pkg_dict = {}
from cocotb import simulator
pkgs = simulator.package_iterate()
if pkgs is None:
packages = SimpleNamespace()
return
for pkg in pkgs:
handle = cast(cocotb.handle.HierarchyObject, cocotb.handle.SimHandle(pkg))
name = handle._name
# Icarus doesn't support named access to package objects:
# https://github.com/steveicarus/iverilog/issues/1038
# so we cannot lazily create handles
if SIM_NAME == "Icarus Verilog":
handle._discover_all()
pkg_dict[name] = handle
packages = SimpleNamespace(**pkg_dict)
def _start_user_coverage() -> None:
if "COVERAGE" in os.environ:
try:
import coverage
except ImportError:
cocotb.log.error(
"Coverage collection requested but coverage module not available. Install it using `pip install coverage`."
)
else:
global _user_coverage
config_filepath = os.getenv("COVERAGE_RCFILE")
if config_filepath is None:
# Exclude cocotb itself from coverage collection.
cocotb.log.info(
"Collecting coverage of user code. No coverage config file supplied via COVERAGE_RCFILE."
)
cocotb_package_dir = os.path.dirname(__file__)
_user_coverage = coverage.coverage(
branch=True, omit=[f"{cocotb_package_dir}/*"]
)
else:
cocotb.log.info(
"Collecting coverage of user code. Coverage config file supplied."
)
# Allow the config file to handle all configuration
_user_coverage = coverage.coverage()
_user_coverage.start()
def _stop_user_coverage() -> None:
if _user_coverage is not None:
_user_coverage.stop()
cocotb.log.debug("Writing coverage data")
_user_coverage.save()
def _setup_random_seed() -> None:
global _random_seed
seed_envvar = os.getenv("RANDOM_SEED")
if seed_envvar is None:
if "ntb_random_seed" in plusargs:
plusarg_seed = plusargs["ntb_random_seed"]
if not isinstance(plusarg_seed, str):
raise TypeError("ntb_random_seed plusarg is not a valid seed value.")
seed = ast.literal_eval(plusarg_seed)
if not isinstance(seed, int):
raise TypeError("ntb_random_seed plusargs is not a valid seed value.")
_random_seed = seed
elif "seed" in plusargs:
plusarg_seed = plusargs["seed"]
if not isinstance(plusarg_seed, str):
raise TypeError("seed plusarg is not a valid seed value.")
seed = ast.literal_eval(plusarg_seed)
if not isinstance(seed, int):
raise TypeError("seed plusargs is not a valid seed value.")
_random_seed = seed
else:
_random_seed = int(time.time())
log.info("Seeding Python random module with %d" % (_random_seed))
else:
_random_seed = ast.literal_eval(seed_envvar)
log.info("Seeding Python random module with supplied seed %d" % (_random_seed))
random.seed(_random_seed)
def _setup_root_handle() -> None:
root_name = os.getenv("TOPLEVEL")
if root_name is not None:
root_name = root_name.strip()
if root_name == "":
root_name = None
elif "." in root_name:
# Skip any library component of the toplevel
root_name = root_name.split(".", 1)[1]
from cocotb import simulator
handle = simulator.get_root_handle(root_name)
if not handle:
raise RuntimeError(f"Can not find root handle {root_name!r}")
global top
top = cocotb.handle.SimHandle(handle)
def _setup_regression_manager() -> None:
global regression_manager
regression_manager = RegressionManager()
# discover tests
module_str = os.getenv("MODULE", "").strip()
if not module_str:
raise RuntimeError(
"Environment variable MODULE, which defines the module(s) to execute, is not defined or empty."
)
modules = [s.strip() for s in module_str.split(",") if s.strip()]
regression_manager.setup_pytest_assertion_rewriting()
regression_manager.discover_tests(*modules)
# filter tests
testcase_str = os.getenv("TESTCASE", "").strip()
test_filter_str = os.getenv("COCOTB_TEST_FILTER", "").strip()
if testcase_str and test_filter_str:
raise RuntimeError("Specify only one of TESTCASE or COCOTB_TEST_FILTER")
elif testcase_str:
warnings.warn(
"TESTCASE is deprecated in favor of COCOTB_TEST_FILTER",
DeprecationWarning,
stacklevel=2,
)
filters = [f"{s.strip()}$" for s in testcase_str.split(",") if s.strip()]
regression_manager.add_filters(*filters)
regression_manager.set_mode(RegressionMode.TESTCASE)
elif test_filter_str:
regression_manager.add_filters(test_filter_str)
regression_manager.set_mode(RegressionMode.TESTCASE)