"""Tests for the AI data analysis agent."""