def main(): depths = [] ranges = [] line = f.readline().rstrip() while line: parts = line.split(': ') depths.append(int(parts[0])) ranges.append(int(parts[1])) line = f.readline().rstrip() firewall = Firewall(depths, ranges) firewall.run() print(firewall.totalSeverity)
def test_severity_with_delay_3(): depths = [0, 1, 4, 6] ranges = [3, 2, 4, 4] firewall = Firewall(depths, ranges, 3) firewall.run() assert firewall.totalSeverity > 0
def test_severity(): depths = [0, 1, 4, 6] ranges = [3, 2, 4, 4] firewall = Firewall(depths, ranges) firewall.run() assert firewall.totalSeverity == 24