Skip to content

Commit

Permalink
Add Sample functionality in DataFrame.
Browse files Browse the repository at this point in the history
Signed-off-by: Vatsal <vatsal.v.anand@gmail.com>
  • Loading branch information
imvtsl committed Nov 2, 2024
1 parent 43b5c77 commit 7779c3d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions internal/tests/integration/dataframe_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -797,7 +797,7 @@ func TestDataFrame_SampleWithReplacement(t *testing.T) {
}
}

func TestDataFrame_SampleWithSeed(t *testing.T) {
func TestDataFrame_SampleSeed(t *testing.T) {
ctx, spark := connect()
df, err := spark.Sql(ctx, "select * from range(100)")
assert.NoError(t, err)
Expand Down Expand Up @@ -831,7 +831,7 @@ func TestDataFrame_SampleWithSeed(t *testing.T) {
assert.Equal(t, rows, rows2)
}

func TestDataFrame_SampleWithReplacementWithSeed(t *testing.T) {
func TestDataFrame_SampleWithReplacementSeed(t *testing.T) {
ctx, spark := connect()
df, err := spark.Sql(ctx, "select * from range(100)")
assert.NoError(t, err)
Expand Down

0 comments on commit 7779c3d

Please sign in to comment.