diff --git a/mglib/mime.py b/mglib/mime.py index 73e001b..7e7845d 100644 --- a/mglib/mime.py +++ b/mglib/mime.py @@ -1,5 +1,5 @@ import logging -from pmworker import wrapper +from . import wrapper logger = logging.getLogger(__name__) diff --git a/setup.py b/setup.py index ffbfcff..953afa1 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ with open("README.md", "r") as fh: setup( name="mglib", - version="1.2.4", + version="1.2.5", author="Eugen Ciur", author_email="eugen@papermerge.com", url="https://github.com/papermerge/mglib", diff --git a/test/test_mime.py b/test/test_mime.py new file mode 100644 index 0000000..06c0929 --- /dev/null +++ b/test/test_mime.py @@ -0,0 +1,23 @@ +import os +from pathlib import Path +import unittest +from mglib import mime + +DATA_DIR = os.path.join( + Path(__file__).parent, + 'data' +) + + +class TestConvert(unittest.TestCase): + + def test_basic_command_call(self): + + file_path = os.path.join( + DATA_DIR, + "berlin.pdf" + ) + mime_type = mime.Mime(filepath=file_path) + self.assertTrue( + mime_type.is_pdf() + )