Add some test cases
stle fix
Merge pull request #9 from AreKai/add_test_cases

Add test cases