input_configuration refactor

christopher-wild · christopher-wild · commit 71b6d215270b · 2023-01-27T10:29:56.000Z
diff --git a/causal_testing/generation/abstract_causal_test_case.py b/causal_testing/generation/abstract_causal_test_case.py
@@ -229,8 +229,8 @@ def generate_concrete_tests(
                     for var in effect_modifier_configs.columns
                 }
             )
-            control_values = [test.control_input_configuration[self.treatment_variable] for test in concrete_tests]
-            treatment_values = [test.treatment_input_configuration[self.treatment_variable] for test in concrete_tests]
+            control_values = [test.control_value for test in concrete_tests]
+            treatment_values = [test.treatment_value for test in concrete_tests]
 
             if self.treatment_variable.datatype is bool and set([(True, False), (False, True)]).issubset(
                 set(zip(control_values, treatment_values))

Original file line number	Diff line number	Diff line change
`@@ -229,8 +229,8 @@ def generate_concrete_tests(`
`229`	`229`	`for var in effect_modifier_configs.columns`
`230`	`230`	`}`
`231`	`231`	`)`
`232`		`- control_values = [test.control_input_configuration[self.treatment_variable] for test in concrete_tests]`
`233`		`- treatment_values = [test.treatment_input_configuration[self.treatment_variable] for test in concrete_tests]`
	`232`	`+ control_values = [test.control_value for test in concrete_tests]`
	`233`	`+ treatment_values = [test.treatment_value for test in concrete_tests]`
`234`	`234`
`235`	`235`	`if self.treatment_variable.datatype is bool and set([(True, False), (False, True)]).issubset(`
`236`	`236`	`set(zip(control_values, treatment_values))`