tests for JPEG and PDF

tests/test_jpeg_format.py

+import unittest
+from formats.jpeg_format import is_valid
+
+
+class TestJpegFormat(unittest.TestCase):
+    def test_is_valid__valid(self):
+        resource = {}
+        with open("tests/data/image.jpeg", "r") as file:
+            self.assertTrue(is_valid(resource, file))
+
+    def test_is_valid__invalid(self):
+        resource = {}
+        with open("tests/data/image.png", "r") as file:
+            self.assertFalse(is_valid(resource, file))
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/test_pdf_format.py

+import unittest
+from formats.pdf_format import is_valid
+
+
+class TestPdfFormat(unittest.TestCase):
+    def test_is_valid__valid(self):
+        resource = {}
+        with open("tests/data/valid.pdf", "r") as file:
+            self.assertTrue(is_valid(resource, file))
+
+    def test_is_valid__invalid(self):
+        resource = {}
+        with open("tests/data/image.png", "r") as file:
+            self.assertFalse(is_valid(resource, file))
+
+
+if __name__ == "__main__":
+    unittest.main()