Skip to content

Commit 7db5810

Browse files
authored
Fix unstable overwrite test (#19)
1 parent 9fee9b6 commit 7db5810

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

paimon_python_java/tests/test_write_and_read.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -310,8 +310,8 @@ def test_overwrite(self):
310310
table_commit = write_builder.new_commit()
311311

312312
df0 = pd.DataFrame({
313-
'f0': [1, 1, 2, 2],
314-
'f1': ['apple', 'banana', 'dog', 'cat'],
313+
'f0': [1, 2],
314+
'f1': ['apple', 'banana'],
315315
})
316316

317317
table_write.write_pandas(df0)
@@ -342,10 +342,10 @@ def test_overwrite(self):
342342

343343
table_scan = read_builder.new_scan()
344344
table_read = read_builder.new_read()
345-
actual_df1 = table_read.to_pandas(table_scan.plan().splits())
345+
actual_df1 = table_read.to_pandas(table_scan.plan().splits()).sort_values(by='f0')
346346
expected_df1 = pd.DataFrame({
347-
'f0': [2, 2, 1],
348-
'f1': ['dog', 'cat', 'watermelon']
347+
'f0': [1, 2],
348+
'f1': ['watermelon', 'banana']
349349
})
350350
expected_df1['f0'] = expected_df1['f0'].astype('int32')
351351
pd.testing.assert_frame_equal(

0 commit comments

Comments
 (0)