debug unittest

AllenInstitute · Oct 10, 2023 · 4160632 · 4160632
1 parent 691333e
commit 4160632
Showing 1 changed file with 3 additions and 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
@@ -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'
+        assert warped.dtype == 'uint8'