Various corrections

Author: Piratmac

2015/24-It Hangs in the Balance.py

@@ -7,7 +7,8 @@
 test_data = {}
 
 test = 1
-test_data[test]   = {"input": """1
+test_data[test] = {
+    "input": """1
 2
 3
 4
@@ -17,26 +18,31 @@
 9
 10
 11""",
-                     "expected": ['99', 'Unknown'],
-                    }
-
-test = 'real'
-input_file = os.path.join(os.path.dirname(__file__), 'Inputs', os.path.basename(__file__).replace('.py', '.txt'))
-test_data[test] = {"input": open(input_file, "r+").read().strip(),
-                     "expected": ['11846773891', 'Unknown'],
-                    }
+    "expected": ["99", "Unknown"],
+}
+
+test = "real"
+input_file = os.path.join(
+    os.path.dirname(__file__),
+    "Inputs",
+    os.path.basename(__file__).replace(".py", ".txt"),
+)
+test_data[test] = {
+    "input": open(input_file, "r+").read().strip(),
+    "expected": ["11846773891", "80393059"],
+}
 
 # -------------------------------- Control program execution -------------------------------- #
 
-case_to_test  = 'real'
-part_to_test  = 2
+case_to_test = "real"
+part_to_test = 2
 verbose_level = 1
 
 # -------------------------------- Initialize some variables -------------------------------- #
 
-puzzle_input           = test_data[case_to_test]['input']
-puzzle_expected_result = test_data[case_to_test]['expected'][part_to_test-1]
-puzzle_actual_result   = 'Unknown'
+puzzle_input = test_data[case_to_test]["input"]
+puzzle_expected_result = test_data[case_to_test]["expected"][part_to_test - 1]
+puzzle_actual_result = "Unknown"
 
 
 # -------------------------------- Actual code execution -------------------------------- #
@@ -45,37 +51,35 @@
 
 mini_quantum_entanglement = 10 ** 100
 
-list_packages = [int(x) for x in puzzle_input.split('\n')]
+list_packages = [int(x) for x in puzzle_input.split("\n")]
 total_weight = sum(list_packages)
 group_weight = total_weight // 3 if part_to_test == 1 else total_weight // 4
 
-for group1_size in range (1, len(list_packages) - 2):
-  for group1 in itertools.combinations(list_packages, group1_size):
-    if sum(group1) != group_weight:
-      continue
-    if reduce(mul, group1, 1) >= mini_quantum_entanglement:
-      continue
+for group1_size in range(1, len(list_packages) - 2):
+    for group1 in itertools.combinations(list_packages, group1_size):
+        if sum(group1) != group_weight:
+            continue
+        if reduce(mul, group1, 1) >= mini_quantum_entanglement:
+            continue
 
-    remaining_packages = [x for x in list_packages if x not in group1]
+        remaining_packages = [x for x in list_packages if x not in group1]
 
-    for group2_size in range (1, len(remaining_packages) - 2):
-      for group2 in itertools.combinations(remaining_packages, group2_size):
-        if sum(group2) == group_weight:
-          mini_quantum_entanglement = min(mini_quantum_entanglement, reduce(mul, group1, 1))
+        for group2_size in range(1, len(remaining_packages) - 2):
+            for group2 in itertools.combinations(remaining_packages, group2_size):
+                if sum(group2) == group_weight:
+                    mini_quantum_entanglement = min(
+                        mini_quantum_entanglement, reduce(mul, group1, 1)
+                    )
 
-  if mini_quantum_entanglement != 10 ** 100:
-    break
+    if mini_quantum_entanglement != 10 ** 100:
+        break
 
 puzzle_actual_result = mini_quantum_entanglement
 
 
 # -------------------------------- Outputs / results -------------------------------- #
 
 if verbose_level >= 3:
-  print ('Input : ' + puzzle_input)
-print ('Expected result : ' + str(puzzle_expected_result))
-print ('Actual result   : '   + str(puzzle_actual_result))
-
-
-
-
+    print("Input : " + puzzle_input)
+print("Expected result : " + str(puzzle_expected_result))
+print("Actual result   : " + str(puzzle_actual_result))

2016/02-Bathroom Security.py

@@ -4,101 +4,103 @@
 test_data = {}
 
 test = 1
-test_data[test]   = {"input": """ULL
+test_data[test] = {
+    "input": """ULL
 RRDDD
 LURDL
 UUUUD""",
-                     "expected": ['1985', '5DB3'],
-                    }
+    "expected": ["1985", "5DB3"],
+}
 
 test += 1
-test_data[test]   = {"input": """""",
-                     "expected": ['Unknown', 'Unknown'],
-                    }
-
-test = 'real'
-input_file = os.path.join(os.path.dirname(__file__), 'Inputs', os.path.basename(__file__).replace('.py', '.txt'))
-test_data[test] = {"input": open(input_file, "r+").read().strip(),
-                     "expected": ['36629', 'Unknown'],
-                    }
+test_data[test] = {
+    "input": """""",
+    "expected": ["Unknown", "Unknown"],
+}
+
+test = "real"
+input_file = os.path.join(
+    os.path.dirname(__file__),
+    "Inputs",
+    os.path.basename(__file__).replace(".py", ".txt"),
+)
+test_data[test] = {
+    "input": open(input_file, "r+").read().strip(),
+    "expected": ["36629", "99C3D"],
+}
 
 # -------------------------------- Control program execution -------------------------------- #
 
-case_to_test  = 'real'
-part_to_test  = 2
+case_to_test = "real"
+part_to_test = 2
 verbose_level = 1
 
 # -------------------------------- Initialize some variables -------------------------------- #
 
-puzzle_input           = test_data[case_to_test]['input']
-puzzle_expected_result = test_data[case_to_test]['expected'][part_to_test-1]
-puzzle_actual_result   = 'Unknown'
+puzzle_input = test_data[case_to_test]["input"]
+puzzle_expected_result = test_data[case_to_test]["expected"][part_to_test - 1]
+puzzle_actual_result = "Unknown"
 
 
 # -------------------------------- Actual code execution -------------------------------- #
 
-password = ''
+password = ""
 
 if part_to_test == 1:
-  keypad = '''123
+    keypad = """123
 456
-789'''
+789"""
 
-  x = 1
-  y = 1
-  for string in puzzle_input.split('\n'):
-    for letter in string:
-      if letter == 'U':
-        y = max(0, y-1)
-      elif letter == 'D':
-        y = min(2, y+1)
-      elif letter == 'L':
-        x = max(0, x-1)
-      elif letter == 'R':
-        x = min(2, x+1)
+    x = 1
+    y = 1
+    for string in puzzle_input.split("\n"):
+        for letter in string:
+            if letter == "U":
+                y = max(0, y - 1)
+            elif letter == "D":
+                y = min(2, y + 1)
+            elif letter == "L":
+                x = max(0, x - 1)
+            elif letter == "R":
+                x = min(2, x + 1)
 
-    password += keypad.split('\n')[y][x]
+        password += keypad.split("\n")[y][x]
 
-  puzzle_actual_result = password
+    puzzle_actual_result = password
 
 
 else:
-  keypad = '''__1__
+    keypad = """__1__
 _234_
 56789
 _ABC_
-__D__'''
-
-  x = 0
-  y = 2
-  for string in puzzle_input.split('\n'):
-    for letter in string:
-      x_new, y_new = x, y
-      if letter == 'U':
-        y_new = max(0, y_new-1)
-      elif letter == 'D':
-        y_new = min(4, y_new+1)
-      elif letter == 'L':
-        x_new = max(0, x_new-1)
-      elif letter == 'R':
-        x_new = min(4, x_new+1)
+__D__"""
 
-      if not keypad.split('\n')[y_new][x_new] == '_':
-        x, y = x_new, y_new
+    x = 0
+    y = 2
+    for string in puzzle_input.split("\n"):
+        for letter in string:
+            x_new, y_new = x, y
+            if letter == "U":
+                y_new = max(0, y_new - 1)
+            elif letter == "D":
+                y_new = min(4, y_new + 1)
+            elif letter == "L":
+                x_new = max(0, x_new - 1)
+            elif letter == "R":
+                x_new = min(4, x_new + 1)
 
-    password += keypad.split('\n')[y][x]
+            if not keypad.split("\n")[y_new][x_new] == "_":
+                x, y = x_new, y_new
 
-  puzzle_actual_result = password
+        password += keypad.split("\n")[y][x]
 
+    puzzle_actual_result = password
 
 
 # -------------------------------- Outputs / results -------------------------------- #
 
 if verbose_level >= 3:
-  print ('Input : ' + puzzle_input)
-print ('Expected result : ' + str(puzzle_expected_result))
-print ('Actual result   : '   + str(puzzle_actual_result))
-
-
-
-
+    print("Input : " + puzzle_input)
+print("Expected result : " + str(puzzle_expected_result))
+print("Actual result   : " + str(puzzle_actual_result))

2016/03-Squares With Three Sides.py

@@ -4,74 +4,75 @@
 test_data = {}
 
 test = 1
-test_data[test]   = {"input": """5 10 25
+test_data[test] = {
+    "input": """5 10 25
 10 15 12""",
-                     "expected": ['Unknown', 'Unknown'],
-                    }
+    "expected": ["Unknown", "Unknown"],
+}
 
 test += 1
-test_data[test]   = {"input": """""",
-                     "expected": ['Unknown', 'Unknown'],
-                    }
-
-test = 'real'
-input_file = os.path.join(os.path.dirname(__file__), 'Inputs', os.path.basename(__file__).replace('.py', '.txt'))
-test_data[test] = {"input": open(input_file, "r+").read().strip(),
-                     "expected": ['983', 'Unknown'],
-                    }
+test_data[test] = {
+    "input": """""",
+    "expected": ["Unknown", "Unknown"],
+}
+
+test = "real"
+input_file = os.path.join(
+    os.path.dirname(__file__),
+    "Inputs",
+    os.path.basename(__file__).replace(".py", ".txt"),
+)
+test_data[test] = {
+    "input": open(input_file, "r+").read().strip(),
+    "expected": ["983", "1836"],
+}
 
 # -------------------------------- Control program execution -------------------------------- #
 
-case_to_test  = 'real'
-part_to_test  = 2
+case_to_test = "real"
+part_to_test = 2
 verbose_level = 1
 
 # -------------------------------- Initialize some variables -------------------------------- #
 
-puzzle_input           = test_data[case_to_test]['input']
-puzzle_expected_result = test_data[case_to_test]['expected'][part_to_test-1]
-puzzle_actual_result   = 'Unknown'
+puzzle_input = test_data[case_to_test]["input"]
+puzzle_expected_result = test_data[case_to_test]["expected"][part_to_test - 1]
+puzzle_actual_result = "Unknown"
 
 
 # -------------------------------- Actual code execution -------------------------------- #
 
 possible_triangles = 0
 if part_to_test == 1:
-  for string in puzzle_input.split('\n'):
-    sides = [int(x) for x in string.split(' ') if not x == '']
-    sides.sort()
-    a, b, c = sides
-
-    if c < (a + b):
-      possible_triangles += 1
+    for string in puzzle_input.split("\n"):
+        sides = [int(x) for x in string.split(" ") if not x == ""]
+        sides.sort()
+        a, b, c = sides
 
+        if c < (a + b):
+            possible_triangles += 1
 
-  puzzle_actual_result = possible_triangles
+    puzzle_actual_result = possible_triangles
 
 else:
-  lines = puzzle_input.split('\n')
-  for n in range(len(lines)):
-    lines[n] = [int(x) for x in lines[n].split(' ') if not x == '']
-  for n in range(len(lines)//3):
-    for i in range (3):
-      sides = [int(lines[n*3+y][i]) for y in range (3)]
-      sides.sort()
-      a, b, c = sides
+    lines = puzzle_input.split("\n")
+    for n in range(len(lines)):
+        lines[n] = [int(x) for x in lines[n].split(" ") if not x == ""]
+    for n in range(len(lines) // 3):
+        for i in range(3):
+            sides = [int(lines[n * 3 + y][i]) for y in range(3)]
+            sides.sort()
+            a, b, c = sides
 
-      if c < (a + b):
-        possible_triangles += 1
-
-  puzzle_actual_result = possible_triangles
+            if c < (a + b):
+                possible_triangles += 1
 
+    puzzle_actual_result = possible_triangles
 
 
 # -------------------------------- Outputs / results -------------------------------- #
 
 if verbose_level >= 3:
-  print ('Input : ' + puzzle_input)
-print ('Expected result : ' + str(puzzle_expected_result))
-print ('Actual result   : '   + str(puzzle_actual_result))
-
-
-
-
+    print("Input : " + puzzle_input)
+print("Expected result : " + str(puzzle_expected_result))
+print("Actual result   : " + str(puzzle_actual_result))