def test_collapse_returns_sum_of_a_level_and_extended_upper_level(self):
pyr = Pyramid(12, 3)
level_0 = PyramidLevel(np.array([[2, 2], [3, 3]], dtype=np.float64), 2,
2)
level_1 = PyramidLevel(np.array([[0]], dtype=np.float64), 1, 1)
pyr.add_lower_resolution_level(level_0)
pyr.add_lower_resolution_level(level_1)
result = pyr.collapse()
self.assertIn(result, level_0.get_array())
def test_collapse_returns_an_array_of_the_same_size_as_the_bottom_level(
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)
pyr.sort_levels()
result = pyr.collapse()
self.assertEqual(result.shape, level_0.get_array().shape)