-
Notifications
You must be signed in to change notification settings - Fork 245
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix offsets for Binary/Lists/LargeLists #727
Conversation
gsilvestrin
commented
Mar 23, 2023
•
edited
Loading
edited
- Added unit tests
- Testes against the data sets of issue Error when querying a Lance dataset #720
|
||
// Make sure the big array was not written to the file | ||
let file_size_bytes = store.size(&path).await.unwrap(); | ||
assert!(file_size_bytes < 1_000); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
seems we just need to make 2000 values instead of 1M?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'm just writing an array that is so big that it will be obvious if the assert doesn't match (I think the actual file size is ~ 250 bytes)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yea, i mean 2000 elements is good enough for test. and it is faster in CI
rust/src/io/reader.rs
Outdated
let large_list_array = LargeListArray::from_iter_primitive::<Int32Type, _, _>(vec![ | ||
Some(vec![Some(10), Some(11)]), | ||
Some(vec![Some(12), Some(13)]), | ||
Some((0..1_000_000).map(|n| Some(n)).collect::<Vec<_>>()), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do we have a test to actually write multiple arrays, wanna just make sure the cummulation of offsets is correct
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not directly, but test_read_struct_of_list_arrays
creates multiple structs that contains lists and write_struct_array
should group them together.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM.
Pending reduce the # of elements created in the test to speed up tests.
|
||
// Make sure the big array was not written to the file | ||
let file_size_bytes = store.size(&path).await.unwrap(); | ||
assert!(file_size_bytes < 1_000); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yea, i mean 2000 elements is good enough for test. and it is faster in CI