16 lines
505 B
Python
16 lines
505 B
Python
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()
|