# 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")