From 615c3cfd62a61d98a8131dff60221614d3cca917 Mon Sep 17 00:00:00 2001 From: Nathan Keim Date: Fri, 13 May 2016 18:08:02 -0700 Subject: [PATCH] TST: Add check for prediction on empty frames. --- trackpy/tests/test_predict.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/trackpy/tests/test_predict.py b/trackpy/tests/test_predict.py index d6c1817c..8dc7e9bf 100644 --- a/trackpy/tests/test_predict.py +++ b/trackpy/tests/test_predict.py @@ -86,6 +86,14 @@ def test_simple_predict(self): pred.link_df_iter, 0.45) assert all(ll.values == 3) + def test_empty_predict(self): + """Check what happens if there is an empty frame""" + pred = self.predict_class() + empty = pandas.DataFrame({'x': [], 'y': [], 'frame': []}) + ll = get_linked_lengths((self.mkframe(0), self.mkframe(0.25), empty, self.mkframe(0.65)), + pred.link_df_iter, 0.45) + assert all(ll.values == 3) + def test_big_predict(self): Nside = Nside_oversize pred = self.predict_class()