Skip to content

Commit 0714002

Browse files
committed
Updated unit tests per review suggestions
1 parent 1ca35d9 commit 0714002

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

python/tests/test_expr.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,7 @@ def test_alias_with_metadata(df):
376376
),
377377
pytest.param(
378378
# Valid values of acosh must be >= 1.0
379-
functions.round((col("a").abs() + lit(1.0)).abs().acosh(), lit(4)),
379+
functions.round((col("a").abs() + lit(1.0)).acosh(), lit(4)),
380380
pa.array([1.1588, 0.9624, 0.0, None], type=pa.float64()),
381381
id="acosh",
382382
),
@@ -717,7 +717,6 @@ def test_alias_with_metadata(df):
717717
],
718718
)
719719
def test_expr_functions(ctx, function, expected_result):
720-
ctx = SessionContext()
721720
batch = pa.RecordBatch.from_arrays(
722721
[
723722
pa.array([-0.75, 0.5, 0.0, None], type=pa.float64()),
@@ -738,4 +737,4 @@ def test_expr_functions(ctx, function, expected_result):
738737
result = df.collect()
739738

740739
assert len(result) == 1
741-
assert result[0].column(0) == expected_result
740+
assert result[0].column(0).equals(expected_result)

0 commit comments

Comments
 (0)