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()