diff --git a/tests/test_offline.py b/tests/test_offline.py index 9c114023da69e9c134ccf7339a168c6b7b6bfce8..c9fd88f49773a5bd5cd4f5ad8fab4c27ebffb34c 100644 --- a/tests/test_offline.py +++ b/tests/test_offline.py @@ -459,6 +459,16 @@ class TestBranchIndexingMagic(unittest.TestCase): self.assertEqual(3, len(self.events[ak.Array([0, 2, 3])])) +class TestBranchHighLevelAccessor(unittest.TestCase): + def test_tracks_arrays(self): + pos_xy = OFFLINE_FILE.tracks.arrays(["pos_x", "pos_y"]) + n = len(pos_xy) + for i in range(n): + assert np.allclose( + OFFLINE_FILE.tracks.pos_x.tolist()[i], pos_xy.pos_x.tolist()[i] + ) + + class TestUsr(unittest.TestCase): def setUp(self): self.f = OFFLINE_USR