Add unit tests for Yutian Li and Ashley Ng