Moved tests to piston-ai_behavior