Test class AIModule