Skip to content

Commit 4b170fc

Browse files
authored
Revert "Removed the getters from test_causal_test_case.py"
1 parent c260527 commit 4b170fc

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

tests/testing_tests/test_causal_test_case.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,17 @@ def setUp(self) -> None:
3737
treatment_value=1,
3838
)
3939

40+
def test_get_treatment_variable(self):
41+
self.assertEqual(self.causal_test_case.get_treatment_variable(), "A")
42+
43+
def test_get_outcome_variable(self):
44+
self.assertEqual(self.causal_test_case.get_outcome_variable(), "C")
45+
46+
def test_get_treatment_value(self):
47+
self.assertEqual(self.causal_test_case.get_treatment_value(), 1)
48+
49+
def test_get_control_value(self):
50+
self.assertEqual(self.causal_test_case.get_control_value(), 0)
4051

4152
def test_str(self):
4253
self.assertEqual(

0 commit comments

Comments
 (0)