add more tests
better import format
Merge pull request #19 from GuoJing/demo

add more tests