add placeholder

Author: kevinjqliu

pyiceberg/table/update/__init__.py

@@ -494,6 +494,20 @@ def _(
     return base_metadata.model_copy(update={"default_sort_order_id": new_sort_order_id})
 
 
+@_apply_table_update.register(RemoveSnapshotRefUpdate)
+def _(update: RemoveSnapshotRefUpdate, base_metadata: TableMetadata, context: _TableMetadataUpdateContext) -> TableMetadata:
+    context.add_update(update)
+    # TODO: implement me
+    return base_metadata
+
+
+@_apply_table_update.register(RemoveSnapshotsUpdate)
+def _(update: RemoveSnapshotsUpdate, base_metadata: TableMetadata, context: _TableMetadataUpdateContext) -> TableMetadata:
+    context.add_update(update)
+    # TODO: implement me
+    return base_metadata
+
+
 def update_table_metadata(
     base_metadata: TableMetadata,
     updates: Tuple[TableUpdate, ...],

tests/table/test_init.py

@@ -1260,10 +1260,11 @@ def test_table_module_refactoring_backward_compatibility() -> None:
         raise pytest.fail("Importing moved modules should not raise an exception") from exc
 
 
-def test_table_update_has_corresponding_dispatch() -> None:
-    from pyiceberg.table import TableUpdate, _apply_table_update
-    # assert that every TableUpdate has a corresponding `_apply_table_update` dispatch function
-    table_update_class = set(TableUpdate.__origin__.__args__)  # type: ignore
-    dispatch_function_class = set(_apply_table_update.registry.keys())
-    missing_dispatch_function = table_update_class - dispatch_function_class
-    assert len(missing_dispatch_function) == 0, f"Missing dispatch function for {missing_dispatch_function}"
+def test_all_table_updates_have_dispatch() -> None:
+    # ensures that every TableUpdate subclass has a corresponding `_apply_table_update` dispatch function.
+    from pyiceberg.table.update import TableUpdate, _apply_table_update
+
+    table_update_classes = set(TableUpdate.__origin__.__args__)  # type: ignore
+    dispatch_classes = set(_apply_table_update.registry.keys())
+    missing_dispatch = table_update_classes - dispatch_classes
+    assert not missing_dispatch, f"Missing dispatch function for: {missing_dispatch}"