From 640b978ee7de8b794b7d79ecd981c1488e6338fb Mon Sep 17 00:00:00 2001 From: misha Date: Thu, 9 Sep 2021 12:02:46 +0300 Subject: [PATCH] post merge fixes --- rust/tests/checkpoint_writer_test.rs | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/rust/tests/checkpoint_writer_test.rs b/rust/tests/checkpoint_writer_test.rs index cf66c8b73c..1ed59ffa32 100644 --- a/rust/tests/checkpoint_writer_test.rs +++ b/rust/tests/checkpoint_writer_test.rs @@ -21,7 +21,7 @@ async fn write_simple_checkpoint() { cleanup_checkpoint_files(log_path.as_path()); // Load the delta table at version 5 - let table = deltalake::open_table_with_version(table_location, 5) + let mut table = deltalake::open_table_with_version(table_location, 5) .await .unwrap(); @@ -96,7 +96,7 @@ mod fs_common; #[tokio::test] async fn test_checkpoints_with_tombstones() { - let main_branch = true; + let main_branch = false; if main_branch { test_checkpoints_with_tombstones_main().await } else { @@ -137,8 +137,8 @@ async fn test_checkpoints_with_tombstones_map_support() { let (removes1, opt1) = pseudo_optimize(&mut table, 5 * 59 * 1000).await; assert_eq!(table.get_files(), vec![opt1.path.as_str()]); assert_eq!( - table.get_tombstones(), - removes1.iter().collect::>() + table.get_state().all_tombstones(), + &removes1 ); checkpoints::create_checkpoint_from_table(&table) @@ -146,7 +146,7 @@ async fn test_checkpoints_with_tombstones_map_support() { .unwrap(); table.update().await.unwrap(); // make table to read the checkpoint assert_eq!(table.get_files(), vec![opt1.path.as_str()]); - assert_eq!(table.get_tombstones(), Vec::<&Remove>::new()); // stale removes are deleted from the state + assert_eq!(table.get_state().all_tombstones(), &vec![]); // stale removes are deleted from the state } async fn pseudo_optimize(table: &mut DeltaTable, offset_millis: i64) -> (Vec, Add) {