Skip to content
Snippets Groups Projects
Commit 074a5a4a authored by Tamas Gal's avatar Tamas Gal :speech_balloon:
Browse files

Add test for arrays()

parent 9e344285
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment