llvm/mlir/test/python/dialects/linalg/opdsl/doctests.py

# RUN: %PYTHON %s

import doctest
import importlib


def test_module(module_name):
    print(f"--- Testing module: {module_name}")
    m = importlib.import_module(module_name)
    doctest.testmod(m, verbose=True, raise_on_error=True, report=True)


test_module("mlir.dialects.linalg.opdsl.lang.affine")
test_module("mlir.dialects.linalg.opdsl.lang.types")