def METHOD_NAME(self):
    size = 2 ** 28
    weighted_sizes = list(map(
        math.ceil, size * FsMultiLevelDataset.compute_size_weights(5)
    ))
    self.assertEqual(
        [201523393,
         50380849,
         12595213,
         3148804,
         787201],
        weighted_sizes
    )