add tests for ai classes