diff --git a/tests/test_loaders.py b/tests/test_loaders.py index ff63131..cc1546b 100644 --- a/tests/test_loaders.py +++ b/tests/test_loaders.py @@ -190,6 +190,7 @@ def test_multiple_image_slice(self): for i, (xbatch, ybatch) in enumerate(loader): self.assertEqual(xbatch[0].mean(), i+1) self.assertEqual(xbatch[1].mean(), i+1+100) + def test_multiple_image_slice_after_split(self): base_dir = nt.fetch_data('example-01')