xobjects 0.5.4__tar.gz → 0.5.6__tar.gz

{xobjects-0.5.4/xobjects.egg-info → xobjects-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.5.4
+Version: 0.5.6
 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

{xobjects-0.5.4 → xobjects-0.5.6}/tests/test_buffer.py

@@ -95,10 +95,12 @@ def test_free_simple(test_context):
         ch.free_string(offset)
         ch.check()
 
+
 @for_all_test_contexts
 def test_free(test_context):
     class CheckFree(xo.Struct):
         a = xo.Float64
+
     ch = CheckFree(a=5, _context=test_context)
     assert ch._buffer.capacity == 8
     assert ch._buffer.chunks == []
@@ -107,17 +109,17 @@ def test_free(test_context):
     with pytest.raises(ValueError, match="Cannot free outside of buffer"):
         ch._buffer.free(0, 10)
     with pytest.raises(ValueError, match="Cannot free outside of buffer"):
-        ch._buffer.free(7,2)
-    ch._buffer.free(0,4)
+        ch._buffer.free(7, 2)
+    ch._buffer.free(0, 4)
     assert len(ch._buffer.chunks) == 1
     assert ch._buffer.chunks[0].start == 0
     assert ch._buffer.chunks[0].end == 4
-    ch._buffer.free(0,4)  # Does nothing
-    ch._buffer.free(2,4)  # Increases free chunk
+    ch._buffer.free(0, 4)  # Does nothing
+    ch._buffer.free(2, 4)  # Increases free chunk
     assert len(ch._buffer.chunks) == 1
     assert ch._buffer.chunks[0].start == 0
     assert ch._buffer.chunks[0].end == 6
-    ch._buffer.free(7,1)
+    ch._buffer.free(7, 1)
     assert len(ch._buffer.chunks) == 2
     assert ch._buffer.chunks[0].start == 0
     assert ch._buffer.chunks[0].end == 6

{xobjects-0.5.4 → xobjects-0.5.6}/tests/test_hybrid_class.py

@@ -330,3 +330,78 @@ def test_to_json_defaults():
     assert np.all(b_dict.pop("a") == [1, 2, 3])
     assert b_dict.pop("d") == 8
     assert b_dict == {}
+
+
+def test_to_dict_python_vars():
+    class TD(xo.HybridClass):
+        _xofields = {
+            "_a": xo.Float64[3],
+            "_b": xo.Int64,
+        }
+        _skip_in_to_dict = ["_a", "_b"]
+        _store_in_to_dict = ["a", "b", "c"]
+
+        def __init__(self, **kwargs):
+            if "_xobject" in kwargs and kwargs["_xobject"] is not None:
+                self.xoinitialize(**kwargs)
+                return
+            kwargs["_a"] = kwargs.pop("a", [1.0, 2.0, 3.0])
+            kwargs["_b"] = kwargs.pop("b", 0)
+            self._initialize(**kwargs)
+
+        def _initialize(self, **kwargs):
+            # Need to handle non-xofields manually
+            c = kwargs.pop("c", -9)
+            super().__init__(**kwargs)
+            self._c = c
+
+        @property
+        def a(self):
+            return self._a
+
+        @property
+        def b(self):
+            return self._b
+
+        @property
+        def c(self):
+            return self._c
+
+    # Verify that to_dict has all fields, including python-only ones,
+    # for default initialisation
+    td1 = TD()
+    td1_dict = td1.to_dict()
+    assert td1_dict.pop("__class__") == "TD"
+    assert all(td1_dict.pop("a") == [1, 2, 3])
+    assert td1_dict.pop("b") == 0
+    assert td1_dict.pop("c") == -9
+    assert td1_dict == {}
+
+    # Verify that to_dict has all fields, including python-only ones,
+    # for custom initialisation
+    td2 = TD(a=[8, 9, 10], b=40, c=20)
+    td2_dict = td2.to_dict()
+    assert td2_dict.pop("__class__") == "TD"
+    assert all(td2_dict.pop("a") == [8, 9, 10])
+    assert td2_dict.pop("b") == 40
+    assert td2_dict.pop("c") == 20
+    assert td2_dict == {}
+
+    # Verify that from_dict works correctly
+    td2_dict = td2.to_dict()
+    td3 = TD.from_dict(td2_dict)
+    assert all(td3.a == td2.a)
+    assert td3.b == td2.b
+    assert td3.c == td2.c
+
+    # Verify that copy works correctly
+    td4 = td3.copy()
+    assert all(td4.a == td2.a)
+    assert td4.b == td2.b
+    assert td4.c == td2.c
+
+    # Verify that move works correctly
+    td3.move(_context=xo.ContextCpu(omp_num_threads="auto"))
+    assert all(td3.a == td2.a)
+    assert td3.b == td2.b
+    assert td3.c == td2.c

xobjects-0.5.4/tests/test_to_json.py → xobjects-0.5.6/tests/test_to_dict.py

@@ -1,7 +1,7 @@
 import xobjects as xo
 
 
-def test_to_json():
+def test_to_dict():
     class A(xo.Struct):
         a = xo.Float64[:]
         b = xo.Int64
@@ -11,14 +11,14 @@ def test_to_json():
 
     x = A(a=[2, 3], b=1)
     u = Uref(x)
-    v = Uref(*u._to_json())
+    v = Uref(*u._to_dict())
 
     assert v.get().a[0] == 2
     assert v.get().a[1] == 3
     assert v.get().b == 1
 
 
-def test_to_json_array():
+def test_to_dict_array():
     class A(xo.Struct):
         a = xo.Float64[:]
 
@@ -35,7 +35,7 @@ def test_to_json_array():
     a[1] = A(a=[3])
     a[5] = B(c=2, d=1)
 
-    b = AUref(a._to_json())
+    b = AUref(a._to_dict())
 
     assert b[1].a[0] == 3
     assert b[5].d == 1

xobjects-0.5.6/xobjects/_version.py

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

{xobjects-0.5.4 → xobjects-0.5.6}/xobjects/array.py

@@ -690,10 +690,13 @@ class Array(metaclass=MetaArray):
         return [cls._itemtype]
 
     def _to_json(self):
+        raise NameError("`_to_json` has been removed. Use `_to_dict` instead.")
+
+    def _to_dict(self):
         out = []
         for v in self:  # TODO does not support multidimensional arrays
-            if hasattr(v, "_to_json"):
-                vdata = v._to_json()
+            if hasattr(v, "_to_dict"):
+                vdata = v._to_dict()
             else:
                 vdata = v
             if self._has_refs and v is not None:

{xobjects-0.5.4 → xobjects-0.5.6}/xobjects/context_cupy.py

@@ -457,7 +457,11 @@ class ContextCupy(XContext):
 
         extra_include_paths = self.get_installed_c_source_paths()
         include_flags = [f"-I{path}" for path in extra_include_paths]
-        extra_compile_args = (*extra_compile_args, *include_flags, "-DXO_CONTEXT_CUDA")
+        extra_compile_args = (
+            *extra_compile_args,
+            *include_flags,
+            "-DXO_CONTEXT_CUDA",
+        )
 
         module = cupy.RawModule(
             code=specialized_source, options=extra_compile_args

{xobjects-0.5.4 → xobjects-0.5.6}/xobjects/hybrid_class.py

@@ -40,7 +40,6 @@ class _FieldOfDressed:
         if self.isnplikearray:
             self.__get__(container=container)[:] = value
         elif hasattr(value, "_xobject"):  # value is a dressed xobject
-
             # Copy xobject data from value inside self._xobject
             # (unless same memory area or Ref and same buffer,
             #  in the latter case reference mechanism is used)
@@ -340,10 +339,18 @@ class HybridClass(metaclass=MetaHybridClass):
         if _context is None and _buffer is None:
             _context = self._xobject._buffer.context
         # This makes a copy of the xobject
-        xobject = self._XoStruct(
+        new_xobject = self._XoStruct(
             self._xobject, _context=_context, _buffer=_buffer, _offset=_offset
         )
-        return self.__class__(_xobject=xobject)
+        new = self.__class__.__new__(self.__class__)
+        new.__dict__.update(self.__dict__)
+        for kk, vv in new.__dict__.items():
+            if kk == "_xobject":
+                continue
+            if hasattr(vv, "copy"):
+                new.__dict__[kk] = vv.copy()
+        new._xobject = new_xobject
+        return new
 
     @property
     def _buffer(self):
@@ -382,14 +389,17 @@ class HybridClass(metaclass=MetaHybridClass):
         return self._xobject.compile_kernels(*args, **kwargs)
 
     def __repr__(self):
-
         if hasattr(self, "_repr_fields"):
             fnames = self._repr_fields
         else:
             fnames = []
             if hasattr(self, "_add_to_repr"):
                 fnames += self._add_to_repr
-            fnames += [fname for fname in self._fields]
+            fnames += [
+                fname
+                for fname in self._xofields.keys()
+                if not fname.startswith("_")
+            ]
             if hasattr(self, "_skip_in_repr"):
                 fnames = [ff for ff in fnames if ff not in self._skip_in_repr]
 
@@ -398,6 +408,12 @@ class HybridClass(metaclass=MetaHybridClass):
             vv = getattr(self, fname)
             if isinstance(vv, float):
                 vvrepr = f"{vv:.3g}"
+            elif isinstance(vv, np.ndarray):
+                vvrepr = repr(vv)
+                # kill "array("
+                vvrepr = vvrepr.split("(", 1)[1]
+                # kill trailing ")"
+                vvrepr = vvrepr.rsplit(")", 1)[0]
             else:
                 vvrepr = repr(vv)
             args.append(f"{fname}={vvrepr}")

{xobjects-0.5.4 → xobjects-0.5.6}/xobjects/ref.py

@@ -294,10 +294,13 @@ class UnionRef(metaclass=MetaUnionRef):
         return cls._reftypes
 
     def _to_json(self):
+        raise NameError("`_to_json` has been removed. Use `_to_dict` instead.")
+
+    def _to_dict(self):
         v = self.get()
         classname = v.__class__.__name__
-        if hasattr(v, "_to_json"):
-            v = v._to_json()
+        if hasattr(v, "_to_dict"):
+            v = v._to_dict()
         return (classname, v)
 
 

{xobjects-0.5.4 → xobjects-0.5.6}/xobjects/struct.py

@@ -366,15 +366,15 @@ class Struct(metaclass=MetaStruct):
             foffset = offset + cls._fields[index].offset
             Int64._to_buffer(buffer, foffset, data_offset)
 
-    def _to_dict(self):
-        return {field.name: field.__get__(self) for field in self._fields}
-
     def _to_json(self):
+        raise NameError("`_to_json` has been removed. Use `_to_dict` instead.")
+
+    def _to_dict(self):
         out = {}
         for field in self._fields:
             v = field.__get__(self)
-            if hasattr(v, "_to_json"):
-                v = v._to_json()
+            if hasattr(v, "_to_dict"):
+                v = v._to_dict()
             out[field.name] = v
         return out
 

{xobjects-0.5.4 → xobjects-0.5.6/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.5.4
+Version: 0.5.6
 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

{xobjects-0.5.4 → xobjects-0.5.6}/xobjects.egg-info/SOURCES.txt

@@ -18,7 +18,7 @@ tests/test_shared_memory.py
 tests/test_strides.py
 tests/test_string.py
 tests/test_struct.py
-tests/test_to_json.py
+tests/test_to_dict.py
 tests/test_typeutils.py
 tests/test_unionref.py
 xobjects/__init__.py

xobjects-0.5.4/xobjects/_version.py

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