"""Tests for AG-UI implementation."""