diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py index 5224666a656e153e93256db647d0c25906190e8b..b367382c834aa3c1a784cef8e4467475bd35e592 100644 --- a/arraycontext/__init__.py +++ b/arraycontext/__init__.py @@ -53,7 +53,8 @@ from .container.traversal import ( from .impl.pyopencl import PyOpenCLArrayContext from .impl.pytato import PytatoArrayContext -from .pytest import pytest_generate_tests_for_array_contexts +from .pytest import (pytest_generate_tests_for_array_contexts, + pytest_generate_tests_for_pyopencl_array_context) from .loopy import make_loopy_program @@ -82,7 +83,8 @@ __all__ = ( "make_loopy_program", - "pytest_generate_tests_for_array_contexts" + "pytest_generate_tests_for_array_contexts", + "pytest_generate_tests_for_pyopencl_array_context" )