File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -154,6 +154,18 @@ def test_global_query(self):
154
154
self .assertTrue (len (query_framework .data ) > 0 )
155
155
self .assertTrue ((query_framework .data ["test_input" ] > 0 ).all ())
156
156
157
+ with open (self .test_config_path , "r" , encoding = "utf-8" ) as f :
158
+ test_configs = json .load (f )
159
+
160
+ test_config = test_configs ["tests" ][0 ].copy ()
161
+ if "query" in test_config :
162
+ del test_config ["query" ]
163
+
164
+ base_test = query_framework .create_base_test (test_config )
165
+ causal_test = query_framework .create_causal_test (test_config , base_test )
166
+
167
+ self .assertTrue ((causal_test .estimator .df ["test_input" ] > 0 ).all ())
168
+
157
169
query_framework .create_variables ()
158
170
query_framework .create_scenario_and_specification ()
159
171
You can’t perform that action at this time.
0 commit comments