From e6c08674d86c83535c7aa4d29fcdd0e82475b224 Mon Sep 17 00:00:00 2001 From: Jianzhun Du Date: Wed, 23 Oct 2024 11:28:08 -0700 Subject: [PATCH] fix --- tests/integ/test_reduce_describe_query.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/integ/test_reduce_describe_query.py b/tests/integ/test_reduce_describe_query.py index ab3b29cad7..c8419f4e68 100644 --- a/tests/integ/test_reduce_describe_query.py +++ b/tests/integ/test_reduce_describe_query.py @@ -36,7 +36,7 @@ def setup(request, session): # Create from SQL create_from_sql_funcs = [ lambda session: session.sql("SELECT 1 AS a, 2 AS b"), - lambda session: session.sql("SELECT 1 AS a, 2 AS b").select("b"), + lambda session: session.sql("SELECT 1 AS a, 2 AS b").select("a"), lambda session: session.sql("SELECT 1 AS a, 2 AS b").select( "a", lit("2").alias("c") ), @@ -47,7 +47,7 @@ def setup(request, session): lambda session: session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"]), lambda session: session.create_dataframe( [[1, 2], [3, 4]], schema=["a", "b"] - ).select("b"), + ).select("a"), lambda session: session.create_dataframe( [[1, 2], [3, 4]], schema=["a", "b"] ).select("a", lit("2").alias("c")), @@ -60,7 +60,7 @@ def setup(request, session): ).cache_result(), lambda session: session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"]) .cache_result() - .select("b"), + .select("a"), lambda session: session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"]) .cache_result() .select("a", lit("2").alias("c")), @@ -72,7 +72,7 @@ def setup(request, session): .group_by("a") .count(), lambda session: session.create_dataframe([[1, 2], [3, 4]], schema=["a", "b"]).join( - session.sql("SELECT 1 AS a, 2 AS b") + session.sql("SELECT 1 AS a, 2 AS b1"), "a" ), lambda session: session.create_dataframe( [[1, 2], [3, 4]], schema=["a", "b"] @@ -100,7 +100,7 @@ def setup(request, session): ), lambda session: session.sql("SELECT 1 AS a, 2 AS b") .union(session.sql("SELECT 3 AS a, 4 AS b")) - .select("b"), + .select("a"), lambda session: session.sql("SELECT 1 AS a, 2 AS b") .union(session.sql("SELECT 3 AS a, 4 AS b")) .select("a", lit("2").alias("c")),