writing AI tests 2