refactor: Simplify iterator usage and improve test assertions in shuffle storage

Author: lukekim

ballista/core/src/execution_plans/shuffle_writer.rs

@@ -646,8 +646,7 @@ impl ShuffleWriterExec {
 
                 let mut part_locs = Vec::new();
 
-                for (output_partition, entry) in
-                    partition_batches.into_iter().enumerate()
+                for (output_partition, entry) in partition_batches.into_iter().enumerate()
                 {
                     if let Some((batches, _num_batches, _num_rows)) = entry {
                         let (path, stats) = storage

ballista/core/src/shuffle_storage.rs

@@ -871,14 +871,9 @@ mod tests {
         assert_eq!(config.storage_type, ShuffleStorageType::Azure);
         assert_eq!(
             config.base_url,
-            Some(
-                "abfs://mycontainer@myaccount.dfs.core.windows.net/shuffle".to_string()
-            )
-        );
-        assert_eq!(
-            config.azure_config.account,
-            Some("myaccount".to_string())
+            Some("abfs://mycontainer@myaccount.dfs.core.windows.net/shuffle".to_string())
         );
+        assert_eq!(config.azure_config.account, Some("myaccount".to_string()));
         assert_eq!(
             config.azure_config.container,
             Some("mycontainer".to_string())
@@ -893,18 +888,13 @@ mod tests {
         )
         .unwrap();
         assert_eq!(config.storage_type, ShuffleStorageType::Azure);
-        assert_eq!(
-            config.azure_config.account,
-            Some("myaccount".to_string())
-        );
+        assert_eq!(config.azure_config.account, Some("myaccount".to_string()));
     }
 
     #[test]
     fn test_from_type_and_url_s3_invalid_url() {
-        let result = ShuffleStorageConfig::from_type_and_url(
-            ShuffleStorageType::S3,
-            "not-a-url",
-        );
+        let result =
+            ShuffleStorageConfig::from_type_and_url(ShuffleStorageType::S3, "not-a-url");
         assert!(result.is_err());
     }
 }