[ADD] Unit tests