From 418f47bcc4cabbe5392ca3b290f5dd27052842e0 Mon Sep 17 00:00:00 2001 From: William Brawner Date: Mon, 9 Dec 2024 22:22:06 -0700 Subject: [PATCH] Add color printing to template --- src/_template.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/_template.py b/src/_template.py index 7b16e08..6052f8d 100644 --- a/src/_template.py +++ b/src/_template.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 -from util import read_input +from util import print_color, read_input, Color def part1(input_lines): @@ -12,10 +12,14 @@ def part2(input_lines): test_input = read_input("_test") -assert 0 == part1(test_input) +test_input_result = part1(test_input) +print_color(test_input_result, Color.GREY) +assert 0 == test_input_result real_input = read_input() print(part1(real_input)) -assert 0 == part2(test_input) +test_input_result = part2(test_input) +print_color(test_input_result, Color.GREY) +assert 0 == test_input_result print(part2(real_input))