5. Unit tests