diff --git a/tests/integ/modin/sql_counter.py b/tests/integ/modin/sql_counter.py index e08f79781c..bf10782d78 100644 --- a/tests/integ/modin/sql_counter.py +++ b/tests/integ/modin/sql_counter.py @@ -264,15 +264,6 @@ def expects(self, **kwargs): print(query, file=sys.stderr) print("=" * len(title), file=sys.stderr) - title = f"\n{'='*20} SqlCounter Captured Describes {'='*20}\n" - describe_set = set() - print(title, file=sys.stderr) - for query in self.get_describe_queries(): - print(query, file=sys.stderr) - describe_set.add(query) - print("=" * len(title), file=sys.stderr) - print(f'Number of describe queries {len(self.get_describe_queries())} and number of unique describe queries {len(describe_set)}\n', file=sys.stderr) - self.clear() def _get_actual_queries(self): @@ -353,9 +344,6 @@ def actual_union_count(self): def actual_describe_count(self): return len([q for q in self._queries if q.is_describe]) - def get_describe_queries(self): - return [q.sql_text for q in self._queries if q.is_describe] - def get_actual_counts(self): """Retrieve all actual counts so far.""" actual_counts = {} diff --git a/tests/integ/test_query_history.py b/tests/integ/test_query_history.py index acc788b61a..4dd5dd5f7e 100644 --- a/tests/integ/test_query_history.py +++ b/tests/integ/test_query_history.py @@ -74,7 +74,7 @@ def test_query_history_multiple_actions(session): with session.query_history(True) as query_history: df = session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"]) df = df.filter(df.a == 1) - res = df.collect() + df.collect() assert len(query_history.queries) == 3 assert query_history.queries[0].is_describe