Skip to content

Commit 24ce234

Browse files
committed
make fake test data float by default to avoid warnings
1 parent 33ac076 commit 24ce234

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

tests/unit/test_coupling.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -176,12 +176,13 @@ def test_coupling_rdt_bump_cmatrix_compare():
176176

177177
def generate_fake_data(n) -> tfs.TfsDataFrame:
178178
qx, qy = 1.31, 1.32
179-
df = tfs.TfsDataFrame(0,
180-
index=[str(i) for i in range(n)],
181-
columns=[S, f"{ALPHA}{X}", f"{ALPHA}{Y}", f"{BETA}{X}", f"{BETA}{Y}",
182-
f"{PHASE_ADV}{X}", f"{PHASE_ADV}{Y}", "R11", "R12", "R21", "R22"],
183-
headers={f"{TUNE}1": qx, f"{TUNE}2": qy}
184-
)
179+
df = tfs.TfsDataFrame(
180+
0.0,
181+
index=[str(i) for i in range(n)],
182+
columns=[S, f"{ALPHA}{X}", f"{ALPHA}{Y}", f"{BETA}{X}", f"{BETA}{Y}",
183+
f"{PHASE_ADV}{X}", f"{PHASE_ADV}{Y}", "R11", "R12", "R21", "R22"],
184+
headers={f"{TUNE}1": qx, f"{TUNE}2": qy},
185+
)
185186

186187
r = np.random.rand(n)
187188
df[S] = np.linspace(0, n, n)

0 commit comments

Comments
 (0)