improve unittest

Signed-off-by: Qu Xiaofeng <quxiaofeng.at.polyu@gmail.com>