[#8] Unit tests