Skip to content

Commit

Permalink
Fix formatting.
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Mar 31, 2021
1 parent a3eeb08 commit fae2063
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions python/cudf/cudf/tests/test_column.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,11 +97,13 @@ def column_slicing_test(col, offset, size, cast_to_float=False):
# sliced, so the indexes should be different and we must ignore it.
# However, we must compare these as frames, not raw arrays, because
# numpy comparison of categorical values won't work.
assert_eq(pd_series[sl].reset_index(drop=True),
sliced_series.reset_index(drop=True))
assert_eq(
pd_series[sl].reset_index(drop=True),
sliced_series.reset_index(drop=True),
)
else:
assert_eq(np.asarray(pd_series[sl]),
sliced_series.to_array())
assert_eq(np.asarray(pd_series[sl]), sliced_series.to_array())


@pytest.mark.parametrize("offset", [0, 1, 15])
@pytest.mark.parametrize("size", [50, 10, 0])
Expand All @@ -119,6 +121,7 @@ def test_decimal_column_slicing(offset, size, precision, scale):
col = col.astype(cudf.Decimal64Dtype(precision, scale))
column_slicing_test(col, offset, size, True)


@pytest.mark.parametrize(
"data",
[
Expand Down

0 comments on commit fae2063

Please sign in to comment.