Skip to content

Commit c80d78e

Browse files
committed
skip doctests that write simulated data to csv files
1 parent f37d3b3 commit c80d78e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

causalpy/data/simulate_data.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,8 @@ def generate_regression_discontinuity_data(
253253
>>> import pathlib
254254
>>> from causalpy.data.simulate_data import generate_regression_discontinuity_data
255255
>>> df = generate_regression_discontinuity_data(true_treatment_threshold=0.5)
256-
>>> df.to_csv(pathlib.Path.cwd() / 'regression_discontinuity.csv', index=False)
256+
>>> df.to_csv(pathlib.Path.cwd() / 'regression_discontinuity.csv',
257+
... index=False) # doctest: +SKIP
257258
"""
258259

259260
def is_treated(x):
@@ -288,7 +289,8 @@ def generate_ancova_data(
288289
... treatment_effect=2,
289290
... sigma=1
290291
... )
291-
>>> df.to_csv(pathlib.Path.cwd() / 'ancova_data.csv', index=False)
292+
>>> df.to_csv(pathlib.Path.cwd() / 'ancova_data.csv',
293+
... index=False) # doctest: +SKIP
292294
"""
293295
group = np.random.choice(2, size=N)
294296
pre = np.random.normal(loc=pre_treatment_means[group])

0 commit comments

Comments
 (0)