Merge pull request #3 from ssl-hep/testing

Update from issue review

Author: ArturU043

README.md

@@ -18,7 +18,7 @@ pip install servicex-analysis-utils
 
     
     Returns:
-        dict: keys are sample names and values are awkward arrays or dask-awkward arrays.
+        dict: keys are sample names and values are awkward arrays, uproot generator objects or dask-awkward arrays.
 ```
 
 ## Documentation

servicex_analysis_utils/materialization.py

@@ -42,7 +42,7 @@ def to_awk(deliver_dict, dask=False, **kwargs):
 
     
     Returns:
-        dict: keys are sample names and values are awkward arrays or dask-awkward arrays.
+        dict: keys are sample names and values are awkward arrays, uproot generator objects or dask-awkward arrays.
     """
 
     awk_arrays = {}
@@ -69,9 +69,7 @@ def to_awk(deliver_dict, dask=False, **kwargs):
             else:
                 if is_root==True:
                     # Use uproot.iterate to handle URLs and local paths files in chunks
-                    tmp_arrays = list(uproot.iterate(paths, library="ak", **kwargs))
-                    # Merge arrays
-                    awk_arrays[sample] = ak.concatenate(tmp_arrays) 
+                    awk_arrays[sample]=uproot.iterate(paths, library="ak", **kwargs) # not an ak array but a generator
                 else:
                     #file is parquet 
                     awk_arrays[sample] = ak.from_parquet(paths, **kwargs)

tests/test_materialization.py

@@ -69,8 +69,8 @@ def test_to_awk(build_test_samples):
 
     #Collecting all samples 
     assert list(result.keys())==["Test-Sample1", "Test-Sample2"]
-    arr1=result["Test-Sample1"]
-    arr2=result["Test-Sample2"]
+    arr1 = ak.concatenate(list(result["Test-Sample1"]))  # Materialize the generator from uproot.iterate  
+    arr2 = ak.concatenate(list(result["Test-Sample2"]))  
 
     #Collecting all branches
     assert ak.fields(arr1) == ['branch1', 'branch2']