xobjects 0.4.0__tar.gz → 0.4.2__tar.gz

{xobjects-0.4.0/xobjects.egg-info → xobjects-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.0
+Version: 0.4.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

{xobjects-0.4.0 → xobjects-0.4.2}/xobjects/__init__.py

@@ -36,6 +36,6 @@ from .linkedarray import BypassLinked
 
 from .general import _print
 
-from .test_helpers import assert_allclose
+from .general import assert_allclose
 
 from ._version import __version__

xobjects-0.4.2/xobjects/_version.py

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

xobjects-0.4.2/xobjects/general.py

@@ -0,0 +1,33 @@
+# copyright ################################# #
+# This file is part of the Xobjects Package.  #
+# Copyright (c) CERN, 2024.                   #
+# ########################################### #
+from numpy.testing import assert_allclose as np_assert_allclose
+import numpy as np
+
+
+class Print:
+    suppress = False
+
+    def __call__(self, *args, **kwargs):
+        if not self.suppress:
+            print(*args, **kwargs)
+
+
+_print = Print()
+
+
+def assert_allclose(a, b, rtol=1e-7, atol=1e-7):
+    if hasattr(a, "get"):
+        a = a.get()
+    if hasattr(b, "get"):
+        b = b.get()
+    try:
+        a = np.squeeze(a)
+    except:
+        pass
+    try:
+        b = np.squeeze(b)
+    except:
+        pass
+    np_assert_allclose(a, b, rtol=rtol, atol=atol)

{xobjects-0.4.0 → xobjects-0.4.2}/xobjects/hybrid_class.py

@@ -46,8 +46,11 @@ class _FieldOfDressed:
             #  in the latter case reference mechanism is used)
             if not (
                 container._xobject._buffer is value._xobject._buffer
-                and (getattr(container._xobject, self.name) is not None and
-                    getattr(container._xobject, self.name)._offset == value._xobject._offset)
+                and (
+                    getattr(container._xobject, self.name) is not None
+                    and getattr(container._xobject, self.name)._offset
+                    == value._xobject._offset
+                )
             ):
                 setattr(container._xobject, self.name, value._xobject)
 
@@ -324,7 +327,8 @@ class HybridClass(metaclass=MetaHybridClass):
 
     @classmethod
     def from_dict(cls, dct, _context=None, _buffer=None, _offset=None):
-        return HybridClass._static_from_dict(cls,
+        return HybridClass._static_from_dict(
+            cls,
             dct,
             _context=_context,
             _buffer=_buffer,

{xobjects-0.4.0 → xobjects-0.4.2}/xobjects/test_helpers.py

@@ -5,7 +5,6 @@
 
 from functools import wraps
 from typing import Callable, Iterable, Union
-from numpy.testing import assert_allclose as np_assert_allclose
 
 import pytest
 
@@ -84,10 +83,3 @@ def requires_context(context_name: str):
         return lambda test_function: test_function
 
     return pytest.mark.skip(f"{context_name} is unavailable on this platform.")
-
-def assert_allclose(a, b, rtol=1e-7, atol=1e-7):
-    if hasattr(a, 'get'):
-        a = a.get()
-    if hasattr(b, 'get'):
-        b = b.get()
-    np_assert_allclose(a, b, rtol=rtol, atol=atol)

{xobjects-0.4.0 → xobjects-0.4.2/xobjects.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xobjects
-Version: 0.4.0
+Version: 0.4.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

xobjects-0.4.0/xobjects/_version.py

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

xobjects-0.4.0/xobjects/general.py

@@ -1,9 +0,0 @@
-class Print:
-    suppress = False
-
-    def __call__(self, *args, **kwargs):
-        if not self.suppress:
-            print(*args, **kwargs)
-
-
-_print = Print()