Adding tests.

@pradeepto please review.
assert should be in caps. Also, you have missed a fullstop.