[DF] Test distributed FromSpec.

Author: martamaja10

python/distrdf/backends/check_fromspec.py

@@ -0,0 +1,27 @@
+import pytest
+
+import ROOT
+
+import DistRDF
+
+class TestFromSpec:
+
+    def test_fromspec(self, payload):
+        """
+        Test usage of FromSpec function
+        """
+        
+        jsonfile = "../data/ttree/spec.json"
+        connection, _ = payload
+        df = ROOT.RDF.Experimental.FromSpec(jsonfile, executor=connection, npartitions=2)
+
+        df_filt = df.Filter("b1 > 42")
+        
+        nentries = 1000
+
+        assert df_filt.Count().GetValue() == nentries
+        assert df_filt.Mean("b1").GetValue() == 50
+
+
+if __name__ == "__main__":
+    pytest.main(args=[__file__])

python/distrdf/backends/test_all.py

@@ -6,6 +6,7 @@
 from check_cloned_actions import *
 from check_distribute_cppcode import * 
 from check_definepersample import *
+from check_fromspec import *
 from check_explicit_api import *
 from check_friend_trees_alignment import *
 from check_friend_trees import *

python/distrdf/data/ttree/_create_datasets.py

@@ -136,3 +136,12 @@ def create_cloned_actions():
                 t.FlushBaskets()
             t.Fill()
         f.WriteObject(t, t.GetName())
+        
+def create_fromspec():
+    # Create a test file for processing
+    treename = "tree"
+    filename = "distrdf_roottest_check_fromspec.root"
+    nentries = 1000
+    opts = ROOT.RDF.RSnapshotOptions()
+    opts.fAutoFlush = 5000
+    ROOT.RDataFrame(nentries).Define("b1", "50").Snapshot(treename, filename, ["b1"], opts)

python/distrdf/data/ttree/distrdf_roottest_check_fromspec.root

@@ -0,0 +1,26 @@
+{
+    "samples": {
+       "sample1": {
+          "trees": [
+             "tree"
+          ],
+          "files": [
+             "../data/ttree/distrdf_roottest_check_rungraphs.root"
+          ],
+          "metadata": {
+             "data_or_mc": "data"
+          }
+       },
+       "sample2": {
+          "trees": [
+             "tree"
+          ],
+          "files": [
+            "../data/ttree/distrdf_roottest_check_fromspec.root"
+          ],
+          "metadata": {
+             "data_or_mc": "mc"
+          }
+       }
+    }
+ }