Add test for entity AI constructors