Clean up assert statement

christopher-wild · christopher-wild · commit 5abe7af9a5a2 · 2023-01-03T10:33:34.000Z
diff --git a/causal_testing/generation/abstract_causal_test_case.py b/causal_testing/generation/abstract_causal_test_case.py
@@ -30,7 +30,7 @@ def __init__(
         effect_modifiers: set[Variable] = None,
         estimate_type: str = "ate",
     ):
-        assert {treatment_variables}.issubset(scenario.variables.values()), (
+        assert (treatment_variables in scenario.variables.values()), (
             "Treatment variables must be a subset of variables."
             + f" Instead got:\ntreatment_variables={treatment_variables}\nvariables={scenario.variables}"
         )

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ def __init__(`
`30`	`30`	`effect_modifiers: set[Variable] = None,`
`31`	`31`	`estimate_type: str = "ate",`
`32`	`32`	`):`
`33`		`- assert {treatment_variables}.issubset(scenario.variables.values()), (`
	`33`	`+ assert (treatment_variables in scenario.variables.values()), (`
`34`	`34`	`"Treatment variables must be a subset of variables."`
`35`	`35`	`+ f" Instead got:\ntreatment_variables={treatment_variables}\nvariables={scenario.variables}"`
`36`	`36`	`)`