Skip to content

Commit ef3e727

Browse files
committed
Merge branch 'fix-deprecation-warnings' of github.com:szymonlopaciuk/xfields into release/0.25.11
2 parents 86175a3 + 1075a77 commit ef3e727

File tree

3 files changed

+15
-11
lines changed

3 files changed

+15
-11
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ include-package-data = true
4848
[project.entry-points.xobjects]
4949
include = "xfields"
5050

51-
[pytest]
51+
[tool.pytest]
5252
markers = [
5353
"context_dependent: marks test as one that depends on the execution context",
5454
]

tests/test_ibs_kicks.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ def test_line_configuration_providing_element(update_every, name):
3737
# Configure and check flags are set as they should be
3838
line.configure_intrabeam_scattering(
3939
element=ibskick,
40-
name=name, # kwarg given to .insert_element()
41-
at_s=100, # kwarg given to .insert_element()
40+
what=name, # kwarg given to .insert_element()
41+
at=100, # kwarg given to .insert_element()
4242
update_every=update_every,
4343
)
4444
assert ibskick.update_every == update_every
@@ -63,7 +63,7 @@ def test_line_configuration_manual_insertion(update_every, name):
6363
# -----------------------------------------------------
6464
# Create element, check flags and manually insert it
6565
ibskick = IBSAnalyticalKick(formalism="b&m", num_slices=50) # same for IBSKineticKick
66-
line.insert_element(element=ibskick, index=-1, name=name)
66+
line.insert(name, obj=ibskick, at=0)
6767
assert ibskick.update_every is None
6868
assert ibskick._name is None
6969
assert ibskick._twiss is None
@@ -93,7 +93,7 @@ def test_configuration_raises_if_not_one_element(num_insertions):
9393
# -----------------------------------------------------
9494
# Inserting element(s) in the line 'num_insertions' times
9595
for i in range(num_insertions):
96-
line.insert_element(element=ibskick, index=-1, name=f"ibskick{i}")
96+
line.insert(f"ibskick{i}", obj=ibskick, at=0)
9797
# -----------------------------------------------------
9898
# Attempting configuration and checking it raises
9999
with pytest.raises(AssertionError):
@@ -120,7 +120,7 @@ def test_configuration_raises_on_below_transition_analytical_kick():
120120
ibskick = IBSAnalyticalKick(formalism="B&M", num_slices=50)
121121
# -----------------------------------------------------
122122
# Insert element in the line and attempt configuration
123-
line.insert_element(element=ibskick, index=10, name="ibskick")
123+
line.insert("ibskick", obj=ibskick, at=line.get_table().rows[10].s[0])
124124
with pytest.raises(AssertionError):
125125
line.configure_intrabeam_scattering(update_every=10)
126126

@@ -147,7 +147,7 @@ def test_kick_coefficients(test_context, formalism):
147147
ibskick = IBSAnalyticalKick(formalism=formalism, num_slices=50)
148148
# -----------------------------------------------------
149149
# Configure in line and generate particles distribution
150-
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", index=0, update_every=1)
150+
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", at=0, update_every=1)
151151
particles = xp.generate_matched_gaussian_bunch(
152152
num_particles=250_000,
153153
nemitt_x=1.2e-5,
@@ -187,7 +187,7 @@ def test_kinetic_coefficients(test_context):
187187
ibskick = IBSKineticKick(num_slices=50)
188188
# -----------------------------------------------------
189189
# Configure in line and generate particles distribution
190-
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", index=0, update_every=1)
190+
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", at=0, update_every=1)
191191
particles = xp.generate_matched_gaussian_bunch(
192192
num_particles=250_000,
193193
nemitt_x=1.2e-5,
@@ -236,7 +236,7 @@ def test_track_analytical_kick(test_context):
236236
cavities = [element for element in line.elements if isinstance(element, xt.Cavity)]
237237
for cavity in cavities:
238238
cavity.lag = 180
239-
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", index=-1, update_every=100)
239+
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", at=0, update_every=100)
240240
tw = line.twiss(method="4d")
241241
particles = xp.generate_matched_gaussian_bunch(
242242
num_particles=2000,
@@ -286,7 +286,7 @@ def test_track_kinetic_kick(test_context):
286286
cavities = [element for element in line.elements if isinstance(element, xt.Cavity)]
287287
for cavity in cavities:
288288
cavity.lag = 180
289-
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", index=-1, update_every=100)
289+
line.configure_intrabeam_scattering(element=ibskick, name="ibskick", at=0, update_every=100)
290290
tw = line.twiss(method="4d")
291291
particles = xp.generate_matched_gaussian_bunch(
292292
num_particles=2000,

xfields/ibs/_api.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
from logging import getLogger
88
from typing import TYPE_CHECKING
9+
from warnings import warn
910

1011
import numpy as np
1112

@@ -175,7 +176,10 @@ def configure_intrabeam_scattering(
175176
line.discard_tracker()
176177
else:
177178
_buffer = None
178-
line.insert_element(element=element, **kwargs)
179+
if 'name' in kwargs:
180+
# Support the previous API that used the deprecated `insert_element(name=...)`
181+
kwargs['what'] = kwargs.pop('name')
182+
line.insert(obj=element, **kwargs)
179183
if _buffer is not None:
180184
line.build_tracker(_buffer=_buffer)
181185
# ----------------------------------------------------------------------------------------------

0 commit comments

Comments
 (0)