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