Add test

@lightsofapollo I added a test codes. Can you confirm?

ref: #9