add assignment tests