"""Tests for AAM Diffusion LLM framework."""