Test compute=False

lincc-frameworks · Mar 28, 2024 · eff0de5 · eff0de5
1 parent d72d75b
commit eff0de5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/tape_tests/test_ensemble_frame.py b/tests/tape_tests/test_ensemble_frame.py
@@ -497,6 +497,8 @@ def test_head(data_fixture, request):
     assert len(frame.head(0)) == 0
     assert len(frame.head(-1)) == 0
 
+    assert len(frame.head(100, compute=False).compute()) == 100
+
     one_res = frame.head(1)
     assert len(one_res) == 1
     assert isinstance(one_res, TapeFrame)