explicitly set logging levels to capture sql

moj-analytical-services · Mar 18, 2024 · 5361a1e · 5361a1e
1 parent bacc765
commit 5361a1e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_join_type_for_estimate_u_and_predict_are_efficient.py b/tests/test_join_type_for_estimate_u_and_predict_are_efficient.py
@@ -175,8 +175,8 @@ def test_link_and_dedupe():
     )
 
     handler.log_list.clear()
-
-    linker.estimate_u_using_random_sampling(max_pairs=1000)
+    logging.getLogger("splink").setLevel(1)
+    linker.estimate_u_using_random_sampling(target_rows=1000)
 
     all_log_messages = "\n".join(log_list)
     all_log_messages = re.sub(r"\s+", " ", all_log_messages)
@@ -231,7 +231,7 @@ def test_link_only_two():
     )
 
     log_list.clear()
-
+    logging.getLogger("splink").setLevel(1)
     linker.estimate_u_using_random_sampling(max_pairs=1000)
 
     all_log_messages = "\n".join(log_list)
@@ -288,7 +288,7 @@ def test_link_only_three():
     )
 
     log_list.clear()
-
+    logging.getLogger("splink").setLevel(1)
     linker.estimate_u_using_random_sampling(max_pairs=1000)
 
     all_log_messages = "\n".join(log_list)