Add unit test [ci skip]