Add CI_Model unit test