Skip to content

Commit

Permalink
Update - remove some unintended changes
Browse files Browse the repository at this point in the history
  • Loading branch information
sfc-gh-evandenberg committed Sep 18, 2024
1 parent 2754839 commit 1937909
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 13 deletions.
12 changes: 0 additions & 12 deletions tests/integ/modin/sql_counter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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 = {}
Expand Down
2 changes: 1 addition & 1 deletion tests/integ/test_query_history.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 1937909

Please sign in to comment.