add tests

Author: chinmay-bhat

tests/table/test_init.py

@@ -652,6 +652,42 @@ def test_update_metadata_add_snapshot(table_v2: Table) -> None:
     assert new_metadata.last_updated_ms == new_snapshot.timestamp_ms
 
 
+def test_update_metadata_rollback_to_snapshot(table_v2: Table) -> None:
+    assert table_v2.rollback_to_snapshot(snapshot_id=3051729675574597004) == Snapshot(
+        snapshot_id=3051729675574597004,
+        parent_snapshot_id=None,
+        sequence_number=0,
+        timestamp_ms=1515100955770,
+        manifest_list='s3://a/b/1.avro',
+        summary=Summary(Operation.APPEND),
+        schema_id=None,
+    )
+
+
+def test_update_metadata_rollback_to_timestamp(table_v2: Table) -> None:
+    assert table_v2.rollback_to_timestamp(timestamp=1555100955771) == Snapshot(
+        snapshot_id=3055729675574597004,
+        parent_snapshot_id=3051729675574597004,
+        sequence_number=1,
+        timestamp_ms=1555100955770,
+        manifest_list='s3://a/b/2.avro',
+        summary=Summary(Operation.APPEND),
+        schema_id=1,
+    )
+
+
+def test_update_metadata_set_current_snapshot(table_v2: Table) -> None:
+    assert table_v2.set_current_snapshot(snapshot_id=3051729675574597004) == Snapshot(
+        snapshot_id=3051729675574597004,
+        parent_snapshot_id=None,
+        sequence_number=0,
+        timestamp_ms=1515100955770,
+        manifest_list='s3://a/b/1.avro',
+        summary=Summary(Operation.APPEND),
+        schema_id=None,
+    )
+
+
 def test_update_metadata_set_snapshot_ref(table_v2: Table) -> None:
     update = SetSnapshotRefUpdate(
         ref_name="main",