From f6aad06bcf8f80bf75dcdfb06d2d820f4211dce8 Mon Sep 17 00:00:00 2001
From: Zineb Aly <zaly@km3net.de>
Date: Tue, 6 Oct 2020 11:30:01 +0200
Subject: [PATCH] add test for rec_stages masks
---
tests/test_tools.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/tests/test_tools.py b/tests/test_tools.py
index 66bac91..8168a3b 100644
--- a/tests/test_tools.py
+++ b/tests/test_tools.py
@@ -183,6 +183,16 @@ class TestRecStagesMasks(unittest.TestCase):
assert masks[1][0] == all(rec_stages[1][0] == ak.Array(stages))
assert masks[0][1] == False
+ def test_mask_with_start_and_end_of_rec_stages(self):
+ rec_stages = self.tracks.rec_stages
+ stages = [1, 3, 5, 4]
+ masks = mask(rec_stages, start=1, end=4)
+
+ assert masks[0][0] == all(rec_stages[0][0] == ak.Array(stages))
+ assert masks[1][0] == all(rec_stages[1][0] == ak.Array(stages))
+ assert masks[0][1] == False
+
+
class TestUnique(unittest.TestCase):
def run_random_test_with_dtype(self, dtype):
--
GitLab