Skip to content

Commit 6c3a574

Browse files
committed
[Python] Update reference refcount values for Python 3.14 compatibility
Follows up on f26668d.
1 parent 49f64d6 commit 6c3a574

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

bindings/pyroot/pythonizations/test/rdataframe_makenumpy.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,14 +83,15 @@ def test_refcount(self):
8383
counts. See also:
8484
https://docs.python.org/3.14/whatsnew/3.14.html#whatsnew314-refcount
8585
"""
86+
extra_ref_count = int(sys.version_info < (3, 14))
8687
data = {"x": np.array([1, 2, 3], dtype="float32")}
8788
gc.collect()
88-
self.assertEqual(sys.getrefcount(data), 1 + int(sys.version_info < (3, 14)))
89+
self.assertEqual(sys.getrefcount(data), 1 + extra_ref_count)
8990
self.assertEqual(sys.getrefcount(data["x"]), 2)
9091

9192
df = ROOT.RDF.FromNumpy(data)
9293
gc.collect()
93-
self.assertEqual(sys.getrefcount(df), 2)
94+
self.assertEqual(sys.getrefcount(df), 1 + extra_ref_count)
9495

9596
self.assertEqual(sys.getrefcount(data["x"]), 3)
9697

bindings/pyroot/pythonizations/test/rvec_asrvec.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,14 +133,15 @@ def test_refcount(self):
133133
numpy array. After deletion of the rvec, the refcount of the numpy array
134134
is decreased.
135135
"""
136+
extra_ref_count = int(sys.version_info < (3, 14))
136137
np_obj = np.array([1, 2])
137138
rvec = ROOT.VecOps.AsRVec(np_obj)
138139
gc.collect()
139-
self.assertEqual(sys.getrefcount(rvec), 1 + int(sys.version_info < (3, 14)))
140-
self.assertEqual(sys.getrefcount(np_obj), 3)
140+
self.assertEqual(sys.getrefcount(rvec), 1 + extra_ref_count)
141+
self.assertEqual(sys.getrefcount(np_obj), 2 + extra_ref_count)
141142
del rvec
142143
gc.collect()
143-
self.assertEqual(sys.getrefcount(np_obj), 2)
144+
self.assertEqual(sys.getrefcount(np_obj), 1 + extra_ref_count)
144145

145146

146147
if __name__ == "__main__":

0 commit comments

Comments
 (0)