Skip to content

Commit 94b6808

Browse files
committed
fix: code cov
1 parent 71b5c5f commit 94b6808

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/main_tests/test_main.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,18 @@ def test_global_query(self):
154154
self.assertTrue(len(query_framework.data) > 0)
155155
self.assertTrue((query_framework.data["test_input"] > 0).all())
156156

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+
157169
query_framework.create_variables()
158170
query_framework.create_scenario_and_specification()
159171

0 commit comments

Comments
 (0)