Set up unit tests

**Is your feature request related to a problem? Please describe.**
We need unit tests to:
- Ensure software quality
- Let developers quickly understand the code logic through unit testing
