Fix bad merge in test_sql_counter.py

snowflakedb · Nov 20, 2024 · eba01a3 · eba01a3
1 parent e952fa0
commit eba01a3
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/integ/modin/test_sql_counter.py b/tests/integ/modin/test_sql_counter.py
@@ -155,27 +155,27 @@ def test_high_sql_count_pass():
 
 def test_sql_count_with_joins():
     with SqlCounter(query_count=1, join_count=1) as sql_counter:
-        sql_counter._add_query(
+        sql_counter._notify(
             QueryRecord(query_id="1", sql_text="SELECT A FROM X JOIN Y")
         )
 
     with SqlCounter(query_count=1, join_count=2) as sql_counter:
-        sql_counter._add_query(
+        sql_counter._notify(
             QueryRecord(query_id="1", sql_text="SELECT A FROM X JOIN Y JOIN Z")
         )
 
     with SqlCounter(query_count=2, join_count=5) as sql_counter:
-        sql_counter._add_query(
+        sql_counter._notify(
             QueryRecord(query_id="1", sql_text="SELECT A FROM X JOIN Y JOIN Z")
         )
-        sql_counter._add_query(
+        sql_counter._notify(
             QueryRecord(query_id="2", sql_text="SELECT A FROM X JOIN Y JOIN Z JOIN W")
         )
 
 
 def test_sql_count_by_query_substr():
     with SqlCounter(query_count=1) as sql_counter:
-        sql_counter._add_query(
+        sql_counter._notify(
             QueryRecord(query_id="1", sql_text="SELECT A FROM X JOIN Y JOIN W")
         )
 
@@ -196,7 +196,7 @@ def test_sql_count_by_query_substr():
 
 def test_sql_count_instances_by_query_substr():
     with SqlCounter(query_count=1) as sql_counter:
-        sql_counter._add_query(
+        sql_counter._notify(
             QueryRecord(query_id="1", sql_text="SELECT A FROM X JOIN Y JOIN W")
         )