fix(pickle): make knowledge picklable #15

ratoaq2 · Jan 19, 2023 · c9b7177 · c9b7177
1 parent 950a2ef
commit c9b7177
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 0 deletions.
diff --git a/knowit/units.py b/knowit/units.py
@@ -25,6 +25,8 @@ def _build_unit_registry():
 
         registry = pint.UnitRegistry()
         registry.define('FPS = 1 * hertz')
+
+        pint.set_application_registry(registry)
         return registry
     except ModuleNotFoundError:
         pass

diff --git a/tests/test_enzyme.py b/tests/test_enzyme.py
@@ -1,3 +1,4 @@
+import pickle
 
 import pytest
 from knowit import KnowitException, know
@@ -19,6 +20,7 @@ def test_enzyme_provider(enzyme, media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
 
 
 @pytest.mark.parametrize('media', mediafiles.get_real_media('enzyme'), ids=id_func)
@@ -36,3 +38,4 @@ def test_enzyme_provider_real_media(media, options):
 
         # Then
         assert_expected(media.expected_data, actual, options)
+        assert pickle.loads(pickle.dumps(actual)) == actual
diff --git a/tests/test_ffmpeg.py b/tests/test_ffmpeg.py
@@ -1,3 +1,4 @@
+import pickle
 
 import pytest
 
@@ -20,6 +21,7 @@ def test_ffmpeg_provider(ffmpeg, media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
 
 
 @pytest.mark.parametrize('media', mediafiles.get_real_media('ffmpeg'), ids=id_func)
@@ -32,3 +34,4 @@ def test_ffmpeg_provider_real_media(media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
diff --git a/tests/test_mediainfo.py b/tests/test_mediainfo.py
@@ -1,3 +1,4 @@
+import pickle
 
 import pytest
 
@@ -17,6 +18,7 @@ def test_mediainfo_provider(mediainfo, media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
 
 
 @pytest.mark.parametrize('media', mediafiles.get_real_media('mediainfo'), ids=id_func)
@@ -29,6 +31,7 @@ def test_mediainfo_provider_real_media(media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
 
 
 @pytest.mark.parametrize('media', mediafiles.get_real_media('mediainfo'), ids=id_func)
@@ -41,3 +44,4 @@ def test_mediainfo_provider_real_media_cli(mediainfo_cli, media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
diff --git a/tests/test_mkvmerge.py b/tests/test_mkvmerge.py
@@ -1,3 +1,4 @@
+import pickle
 
 import pytest
 
@@ -20,6 +21,7 @@ def test_mkvmerge_provider(mkvmerge, media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual
 
 
 @pytest.mark.parametrize('media', mediafiles.get_real_media('mkvmerge'), ids=id_func)
@@ -32,3 +34,4 @@ def test_mkvmerge_provider_real_media(media, options):
 
     # Then
     assert_expected(media.expected_data, actual, options)
+    assert pickle.loads(pickle.dumps(actual)) == actual