fix tests

Author: rahul2393

samples/samples/snippets.py

@@ -1591,9 +1591,9 @@ def __init__(self):
             super().__init__("commit_stats_sample")
 
         def info(self, msg, *args, **kwargs):
-            if kwargs["extra"] and "commit_stats" in kwargs["extra"]:
+            if "extra" in kwargs and kwargs["extra"] and "commit_stats" in kwargs["extra"]:
                 self.last_commit_stats = kwargs["extra"]["commit_stats"]
-            super().info(msg)
+            super().info(msg, *args, **kwargs)
 
     spanner_client = spanner.Client()
     instance = spanner_client.instance(instance_id)

samples/samples/snippets_test.py

@@ -683,7 +683,7 @@ def test_write_with_dml_transaction(capsys, instance_id, sample_database):
 def test_update_data_with_partitioned_dml(capsys, instance_id, sample_database):
     snippets.update_data_with_partitioned_dml(instance_id, sample_database.database_id)
     out, _ = capsys.readouterr()
-    assert "3 record(s) updated" in out
+    assert "3 records updated" in out
 
 
 @pytest.mark.dependency(