xobjects 0.7.1__tar.gz → 0.7.2__tar.gz

{xobjects-0.7.1/xobjects.egg-info → xobjects-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xobjects
-Version: 0.7.1
+Version: 0.7.2
 Summary: In-memory serialization and code generator for CPU and GPU
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xobjects
@@ -17,6 +17,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy
 Requires-Dist: cffi
+Requires-Dist: setuptools
 Requires-Dist: scipy
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"

{xobjects-0.7.1 → xobjects-0.7.2}/pyproject.toml

@@ -17,6 +17,7 @@ requires-python = ">=3.9"
 dependencies = [
     "numpy",
     "cffi",
+    "setuptools",
     "scipy"
 ]
 [project.urls]

{xobjects-0.7.1 → xobjects-0.7.2}/setup.py

@@ -23,7 +23,7 @@ setup(
     url="https://xsuite.readthedocs.io/",
     python_requires=">=3.7",
     setup_requires=[],
-    install_requires=["numpy", "cffi", "scipy"],
+    install_requires=["numpy", "cffi", "setuptools", "scipy"],
     packages=["xobjects"],
     license="Apache 2.0",
     download_url="https://pypi.python.org/pypi/xobjects",

{xobjects-0.7.1 → xobjects-0.7.2}/tests/test_kernel.py

@@ -4,6 +4,7 @@
 # ########################################### #
 
 import cffi
+import pickle
 import pytest
 import sysconfig
 from pathlib import Path
@@ -14,7 +15,8 @@ import xobjects as xo
 from xobjects.test_helpers import for_all_test_contexts, requires_context
 
 
-def test_kernel_cpu():
+@pytest.mark.parametrize('pickle_context', [False, True])
+def test_kernel_cpu(pickle_context):
     ctx = xo.ContextCpu()
     src_code = r"""
 double my_mul(const int n, const double* x1,
@@ -45,6 +47,13 @@ double my_mul(const int n, const double* x1,
 
     assert y == 285.0
 
+    if pickle_context:
+        restored = pickle.loads(pickle.dumps(ctx))
+        assert not restored.kernels
+        assert ctx.kernels.my_mul(n=len(a1), x1=a1, x2=a2) == 285.0
+        restored.add_kernels(sources=[src_code], kernels=kernel_descriptions)
+        assert restored.kernels.my_mul(n=len(a1), x1=a1, x2=a2) == 285.0
+
 
 @for_all_test_contexts
 def test_kernel_pointer_none_maps_to_null(test_context):

xobjects-0.7.2/xobjects/_version.py

@@ -0,0 +1 @@
+__version__ = "0.7.2"

{xobjects-0.7.1 → xobjects-0.7.2}/xobjects/context_cpu.py

@@ -96,6 +96,7 @@ def kernel_compilation_help_message():
 
 from .context import (
     Kernel,
+    KernelDict,
     ModuleNotAvailable,
     SourceType,
     XBuffer,
@@ -821,6 +822,7 @@ class ContextCpu(XContext):
 
     def __setstate__(self, state):
         self.__dict__.update(state)
+        self._kernels = KernelDict(self._kernels)
         self._buffers = weakref.WeakSet()
 
 

{xobjects-0.7.1 → xobjects-0.7.2/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xobjects
-Version: 0.7.1
+Version: 0.7.2
 Summary: In-memory serialization and code generator for CPU and GPU
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xobjects
@@ -17,6 +17,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy
 Requires-Dist: cffi
+Requires-Dist: setuptools
 Requires-Dist: scipy
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"

{xobjects-0.7.1 → xobjects-0.7.2}/xobjects.egg-info/requires.txt

@@ -1,5 +1,6 @@
 numpy
 cffi
+setuptools
 scipy
 
 [tests]

xobjects-0.7.1/xobjects/_version.py

@@ -1 +0,0 @@
-__version__ = "0.7.1"