From ed8bc7daa3050af66ccc766eefc4d96489011d58 Mon Sep 17 00:00:00 2001
From: Stefan Wunsch <stefan.wunsch@cern.ch>
Date: Wed, 13 Mar 2019 14:11:49 +0100
Subject: [PATCH] [PyROOT] Add test for reading empty arrays and selections
with AsNumpy
---
bindings/pyroot/test/rdataframe_asnumpy.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/bindings/pyroot/test/rdataframe_asnumpy.py b/bindings/pyroot/test/rdataframe_asnumpy.py
index 1ffa170ef4b..9532b2ed628 100644
--- a/bindings/pyroot/test/rdataframe_asnumpy.py
+++ b/bindings/pyroot/test/rdataframe_asnumpy.py
@@ -194,6 +194,17 @@ class RDataFrameAsNumpy(unittest.TestCase):
self.assertTrue(all(x == ref))
self.assertTrue(hasattr(x, "result_ptr"))
+ def test_empty_array(self):
+ df = ROOT.RDataFrame(1).Define("x", "std::vector<float>()")
+ npy = df.AsNumpy(["x"])
+ self.assertEqual(npy["x"].size, 1)
+ self.assertTrue(npy["x"][0].empty())
+
+ def test_empty_selection(self):
+ df = ROOT.RDataFrame(10).Define("x", "1.0").Filter("x<0")
+ npy = df.AsNumpy(["x"])
+ self.assertEqual(npy["x"].size, 0)
+
if __name__ == '__main__':
unittest.main()
--
GitLab