diff --git a/tests/integ/modin/test_sql_counter.py b/tests/integ/modin/test_sql_counter.py index cffd826735..048e5bca52 100644 --- 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") )