@@ -61,7 +61,7 @@ class TestDetectionDatasetCaching(unittest.TestCase):
 
         self.assertEqual(1, self._count_cached_array())
         for first_dataset, second_dataset in zip(datasets[:-1], datasets[1:]):
-            self.assertTrue(np.array_equal(first_dataset.cached_imgs, second_dataset.cached_imgs))
+            self.assertTrue(np.array_equal(first_dataset.cached_imgs_padded, second_dataset.cached_imgs_padded))
 
         self._empty_cache()
 
@@ -76,7 +76,7 @@ class TestDetectionDatasetCaching(unittest.TestCase):
 
         self.assertEqual(5, self._count_cached_array())
         for first_dataset, second_dataset in zip(datasets[:-1], datasets[1:]):
-            self.assertFalse(np.array_equal(first_dataset.cached_imgs, second_dataset.cached_imgs))
+            self.assertFalse(np.array_equal(first_dataset.cached_imgs_padded, second_dataset.cached_imgs_padded))
 
         self._empty_cache()