Modify test function
Merge pull request #26 from ZhishenYang/social_media

Modify test function