add tests and minor changes

Author: PengLiVectra

python/deltalake/_internal.pyi

@@ -35,7 +35,6 @@ class RawDeltaTable:
     @staticmethod
     def load_lazy(
         table_uri: str,
-        version: Optional[int],
         storage_options: Optional[Dict[str, str]],
         without_files: bool,
         log_buffer_size: Optional[int],

python/deltalake/table.py

@@ -255,7 +255,6 @@ def __init__(
         if lazy_load:
             self._table = RawDeltaTable.load_lazy(
                 str(table_uri),
-                version=version,
                 storage_options=storage_options,
                 without_files=without_files,
                 log_buffer_size=log_buffer_size,

python/src/lib.rs

@@ -124,11 +124,10 @@ impl RawDeltaTable {
     }
 
     #[classmethod]
-    #[pyo3(signature = (table_uri, version = None, storage_options = None, without_files = false, log_buffer_size = None))]
+    #[pyo3(signature = (table_uri, storage_options = None, without_files = false, log_buffer_size = None))]
     fn load_lazy(
         _cls: &PyType,
         table_uri: &str,
-        version: Option<i64>,
         storage_options: Option<HashMap<String, String>>,
         without_files: bool,
         log_buffer_size: Option<usize>,
@@ -138,9 +137,6 @@ impl RawDeltaTable {
         if let Some(storage_options) = storage_options {
             builder = builder.with_storage_options(storage_options)
         }
-        if let Some(version) = version {
-            builder = builder.with_version(version)
-        }
         if without_files {
             builder = builder.without_files()
         }

python/tests/test_table_read.py

@@ -63,6 +63,25 @@ def test_read_simple_table_using_options_to_dict():
     assert dt.to_pyarrow_dataset().to_table().to_pydict() == {"value": [1, 2, 3]}
 
 
+def test_simple_table_lazy_loading():
+    table_path = "../crates/deltalake-core/tests/data/simple_table"
+    dt = DeltaTable(table_path, lazy_load=True)
+    dt.load_version(2)
+    assert dt.version() == 2
+
+
+def test_simple_table_lazy_loading_with_options():
+    table_path = "../crates/deltalake-core/tests/data/simple_table"
+    dt = DeltaTable(
+        table_path,
+        storage_options={},
+        without_files=False,
+        log_buffer_size=1,
+        lazy_load=True,
+    )
+    assert isinstance(dt, DeltaTable)
+
+
 def test_load_with_datetime():
     log_dir = "../crates/deltalake-core/tests/data/simple_table/_delta_log"
     log_mtime_pair = [