import unittest
from organizer import categorizar_por_tipo
from pathlib import Path


class TestOrganizer(unittest.TestCase):
    def test_categorizar_por_tipo(self):
        self.assertEqual(categorizar_por_tipo("imagem.jpg"), "imagem")
        self.assertEqual(categorizar_por_tipo("documento.pdf"), "documento")
        self.assertEqual(categorizar_por_tipo("video.mp4"), "video")
        self.assertEqual(categorizar_por_tipo("ficheiro.zip"), "outro")


if __name__ == "__main__":
    unittest.main()