xref: /llvm-project/mlir/test/python/dialects/linalg/opdsl/doctests.py (revision f9008e6366c2496b1ca1785b891d5578174ad63e) 
1# RUN: %PYTHON %s
2
3import doctest
4import importlib
5
6
7def test_module(module_name):
8    print(f"--- Testing module: {module_name}")
9    m = importlib.import_module(module_name)
10    doctest.testmod(m, verbose=True, raise_on_error=True, report=True)
11
12
13test_module("mlir.dialects.linalg.opdsl.lang.affine")
14test_module("mlir.dialects.linalg.opdsl.lang.types")
15