def test_sort_levels_sorts_levels_according_to_size_larger_first(self): pyr = Pyramid(12, 3) level_0 = PyramidLevel( np.array([[10, 10, 10], [2, 2, 2], [4, 4, 4]], dtype=np.float64), 2, 2) level_1 = PyramidLevel(np.array([[2, 2], [3, 3]], dtype=np.float64), 2, 2) level_2 = PyramidLevel(np.array([0], dtype=np.float64), 1, 1) pyr.add_lower_resolution_level(level_1) pyr.add_lower_resolution_level(level_0) pyr.add_lower_resolution_level(level_2) self.assertEqual( pyr.get_level(0).get_array().shape, level_1.get_array().shape) pyr.sort_levels() self.assertEqual( pyr.get_level(0).get_array().shape, level_0.get_array().shape)
def test_get_level_returns_the_correct_list_element(self): pyr = Pyramid(10, 3) pyr.levels = [12, 13, 14] self.assertEqual(pyr.get_level(1), 13) self.assertEqual(pyr.get_level(0), 12)