update

tests/test_lib.py

+import pytest
+import tempfile
+import os
+import uuid
+
+from mpai_cae_arp.types.irregularity import Irregularity, Source, IrregularityFile
+
+from tape_irregularity_classifier import lib as tic
+
+
+def test_verify_path():
+
+    with pytest.raises(FileNotFoundError):
+        working_path = ""
+        files_name = ""
+        tic.verify_path(working_path, files_name)
+    
+    with tempfile.TemporaryDirectory() as working_path:
+        files_name = "test"
+        os.makedirs(working_path + "/temp/" + files_name)
+        assert tic.verify_path(working_path, files_name) == working_path + "/temp/" + files_name
+    
+
+def test_collect_irregularity_images():
+    irregularities = []
+    assert tic.collect_irregularity_images(irregularities) == []
+
+    irregularities = [
+        Irregularity(
+            irregularity_ID=uuid.uuid4(),
+            source=Source.VIDEO,
+            time_label="00:00:00.000",
+            image_URI="test1.jpg",
+        )]
+    assert tic.collect_irregularity_images(irregularities) == []
+    assert irregularities == []
+
+
+def test_get_irregularity_file():
+    
+    with pytest.raises(FileNotFoundError):
+        tic.get_irregularity_file("")
+    
\ No newline at end of file