From 4160632d7e13daeed48b6a6cc6ff1d07202d29ce Mon Sep 17 00:00:00 2001 From: Christopher Morrison Date: Tue, 10 Oct 2023 12:47:44 -0700 Subject: [PATCH] debug unittest --- .../behavior_project_cache/test_natural_movie_cache.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/allensdk/test/brain_observatory/behavior/behavior_project_cache/test_natural_movie_cache.py b/allensdk/test/brain_observatory/behavior/behavior_project_cache/test_natural_movie_cache.py index 24950742f..29df03045 100644 --- a/allensdk/test/brain_observatory/behavior/behavior_project_cache/test_natural_movie_cache.py +++ b/allensdk/test/brain_observatory/behavior/behavior_project_cache/test_natural_movie_cache.py @@ -24,13 +24,12 @@ def test_natural_movie_cache(): cache = NaturalMovieOneCache(cache_dir='fake_dir', bucket_name='fake_bucket') movie = cache.get_processed_template_movie(n_workers=1) - import pdb; pdb.set_trace() assert movie.index.name == 'movie_frame_index' assert movie.columns.to_list() == ['unwarped', 'warped'] - unwarped = movie.loc[0, 'unwarped'].shape - warped = movie.loc[0, 'warped'].shape + unwarped = movie.loc[0, 'unwarped'] + warped = movie.loc[0, 'warped'] assert unwarped.shape == (1200, 1920) assert warped.shape == (1200, 1920) assert unwarped.dtype == 'float64' - assert warped.dtype == 'uint8' \ No newline at end of file + assert warped.dtype == 'uint8'