Skip to content

Commit 8e33b25

Browse files
Update tests to use pd.Series for confidence intervals
1 parent 8590dc9 commit 8e33b25

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/testing_tests/test_causal_test_outcome.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def test_Negative_fail_ci(self):
151151
ctr = CausalTestResult(
152152
estimator=self.estimator,
153153
test_value=test_value,
154-
confidence_intervals=[-1, 1],
154+
confidence_intervals=[pd.Series(-1), pd.Series(1)],
155155
effect_modifier_configuration=None,
156156
)
157157
ev = Negative()
@@ -173,7 +173,7 @@ def test_exactValue_pass_ci(self):
173173
ctr = CausalTestResult(
174174
estimator=self.estimator,
175175
test_value=test_value,
176-
confidence_intervals=[4, 6],
176+
confidence_intervals=[pd.Series(4), pd.Series(6)],
177177
effect_modifier_configuration=None,
178178
)
179179
ev = ExactValue(5, 0.1)
@@ -199,7 +199,7 @@ def test_invalid(self):
199199
ctr = CausalTestResult(
200200
estimator=self.estimator,
201201
test_value=test_value,
202-
confidence_intervals=[4.8, 6.7],
202+
confidence_intervals=[pd.Series(4.8), pd.Series(6.7)],
203203
effect_modifier_configuration=None,
204204
)
205205
with self.assertRaises(ValueError):
@@ -216,7 +216,7 @@ def test_someEffect_pass_coefficient(self):
216216
ctr = CausalTestResult(
217217
estimator=self.estimator,
218218
test_value=test_value,
219-
confidence_intervals=[4.8, 6.7],
219+
confidence_intervals=[pd.Series(4.8), pd.Series(6.7)],
220220
effect_modifier_configuration=None,
221221
)
222222
self.assertTrue(SomeEffect().apply(ctr))
@@ -227,7 +227,7 @@ def test_someEffect_pass_ate(self):
227227
ctr = CausalTestResult(
228228
estimator=self.estimator,
229229
test_value=test_value,
230-
confidence_intervals=[4.8, 6.7],
230+
confidence_intervals=[pd.Series(4.8), pd.Series(6.7)],
231231
effect_modifier_configuration=None,
232232
)
233233
self.assertTrue(SomeEffect().apply(ctr))
@@ -238,7 +238,7 @@ def test_someEffect_pass_rr(self):
238238
ctr = CausalTestResult(
239239
estimator=self.estimator,
240240
test_value=test_value,
241-
confidence_intervals=[4.8, 6.7],
241+
confidence_intervals=[pd.Series(4.8), pd.Series(6.7)],
242242
effect_modifier_configuration=None,
243243
)
244244
self.assertTrue(SomeEffect().apply(ctr))
@@ -249,7 +249,7 @@ def test_someEffect_fail(self):
249249
ctr = CausalTestResult(
250250
estimator=self.estimator,
251251
test_value=test_value,
252-
confidence_intervals=[-0.1, 0.2],
252+
confidence_intervals=[pd.Series(-0.1), pd.Series(0.2)],
253253
effect_modifier_configuration=None,
254254
)
255255
self.assertFalse(SomeEffect().apply(ctr))
@@ -260,7 +260,7 @@ def test_someEffect_str(self):
260260
ctr = CausalTestResult(
261261
estimator=self.estimator,
262262
test_value=test_value,
263-
confidence_intervals=[-0.1, 0.2],
263+
confidence_intervals=[pd.Series(-0.1), pd.Series(0.2)],
264264
effect_modifier_configuration=None,
265265
)
266266
ev = SomeEffect()
@@ -284,7 +284,7 @@ def test_someEffect_dict(self):
284284
ctr = CausalTestResult(
285285
estimator=self.estimator,
286286
test_value=test_value,
287-
confidence_intervals=[-0.1, 0.2],
287+
confidence_intervals=[pd.Series(-0.1), pd.Series(0.2)],
288288
effect_modifier_configuration=None,
289289
)
290290
ev = SomeEffect()

0 commit comments

Comments
 (0)