def solve(task): """ Solve the puzzle Args: task (str): length x width x height \n ... (without spaces) Returns: int: Total feet of ribbon """ data = process_data(task) return sum(2 * (sum(size) - max(size)) + reduce(mul, size, 1) for size in data)
def test_returns_correct_result(self): self.assertEqual(process_data('1x2x3'), [(1, 2, 3)]) self.assertEqual(process_data('1x2x3\n4x5x6'), [(1, 2, 3), (4, 5, 6)])