CLN: Remove default name for a new axis

dynamicslab · Jan 6, 2024 · 0ae9f6a · 0ae9f6a
1 parent 341f15c
commit 0ae9f6a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/pysindy/utils/axes.py b/pysindy/utils/axes.py
@@ -242,15 +242,15 @@ def __array_finalize__(self, obj) -> None:
             (
                 not isinstance(obj, AxesArray),
                 self.shape == (),
-                not hasattr(self, "__ax_map"),
+                not hasattr(self, "_ax_map"),
             )
         ):
             self._ax_map = _AxisMapping({})
         # required by ravel() and view() used in numpy testing.  Also for zeros_like...
         elif all(
             (
                 isinstance(obj, AxesArray),
-                not hasattr(self, "__ax_map"),
+                not hasattr(self, "_ax_map"),
                 self.shape == obj.shape,
             )
         ):

diff --git a/test/utils/test_axes.py b/test/utils/test_axes.py
@@ -100,7 +100,7 @@ def __array_ufunc__(self, ufunc, method, *inputs, **kwargs):
     assert_(c is b)
     check = np.add.accumulate(d, axis=0)
     c = np.add.accumulate(a, axis=0)
-    # assert_equal(c, check)
+    assert_equal(c, check)
     b = np.zeros_like(c)
     c = np.add.accumulate(a, 0, None, b)
     assert_equal(c, check)
@@ -119,11 +119,11 @@ def __array_ufunc__(self, ufunc, method, *inputs, **kwargs):
     a = d.copy().view(AxesArray)
     np.add.at(check, ([0, 1], [0, 2]), 1.0)
     np.add.at(a, ([0, 1], [0, 2]), 1.0)
-    assert_equal(a, check)
+    assert_equal(np.asarray(a), np.asarray(check))  # modified
     b = np.array(1.0).view(AxesArray)
     a = d.copy().view(AxesArray)
     np.add.at(a, ([0, 1], [0, 2]), b)
-    assert_equal(a, check)
+    assert_equal(np.asarray(a), np.asarray(check))  # modified
 
 
 def test_n_elements():