Fix bug in Table.__setattr__: don't terminate on cache invalidation

xsuite · Dec 10, 2024 · e3de3d0 · e3de3d0
1 parent dbca11b
commit e3de3d0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -123,6 +123,8 @@ def test_table_setitem_col():
     assert np.array_equal(t["2betx"], data["betx"] * 2)
     t["betx"] = 1
     assert np.array_equal(t["betx"], np.ones(len(data["betx"])))
+    t["name"] = t["name"] * 2
+    assert np.all(t['name'] == [x * 2 for x in data['name']])
 
 
 def test_table_setitem_col_row():

diff --git a/xdeps/table.py b/xdeps/table.py
@@ -829,7 +829,8 @@ def __setitem__(self, key, val):
             object.__setattr__(self, "_index_cache", None)
             object.__setattr__(self, "_count_cache", None)
             object.__setattr__(self, "_name_cache", None)
-        elif isinstance(key, str):
+
+        if isinstance(key, str):
             if key in self.__dict__:
                 object.__setattr__(self, key, val)
             elif key in self._col_names: