fixes

spiraldb · Oct 23, 2024 · 1a4dc16 · 1a4dc16
1 parent c09e5a2
commit 1a4dc16
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/pyvortex/test/test_dataset.py b/pyvortex/test/test_dataset.py
@@ -53,7 +53,7 @@ def test_to_batches(ds):
 
     chunk0 = next(ds.to_batches(columns=["string", "bool"]))
     assert chunk0.to_struct_array() == pa.array(
-        [record(x, columns=["string", "bool"]) for x in range(1 << 16)], type=schema
+        [record(x, columns=["string", "bool"]) for x in range(1_000_000)], type=schema
     )
 
 

diff --git a/vortex-serde/src/layouts/read/buffered.rs b/vortex-serde/src/layouts/read/buffered.rs
@@ -100,9 +100,9 @@ impl BufferedArrayReader {
                     .front()
                     .map(|((begin, _), _)| up_to_row < *begin)
             })
-            .unwrap_or(true)
+            .unwrap_or(false)
         {
-            vortex_bail!("Can't advance backwards")
+            vortex_bail!("Can't advance backwards {up_to_row}")
         }
 
         let mut new_arrays = mem::take(&mut self.arrays)
@@ -252,9 +252,9 @@ impl BufferedSelectorReader {
                     .front()
                     .map(|((begin, _), _)| up_to_row < *begin)
             })
-            .unwrap_or(true)
+            .unwrap_or(false)
         {
-            vortex_bail!("Can't advance backwards")
+            vortex_bail!("Can't advance backwards to {up_to_row}")
         }
 
         let mut new_selectors = mem::take(&mut self.selectors)