def test_expansions(self): expansions = load_map(get_test_file_path("samples/d14.txt")) self.assertEqual(apply_expansions(expansions, "NNCB"), "NCNBCHB") self.assertEqual(apply_expansions(expansions, "NCNBCHB"), "NBCCNBBBCBHCB") self.assertEqual(apply_expansions(expansions, "NBCCNBBBCBHCB"), "NBBBCNCCNBBNBNBBCHBHHBCHB") self.assertEqual( apply_expansions(expansions, "NBBBCNCCNBBNBNBBCHBHHBCHB"), "NBBNBNBBCCNBCNCCNBBNBBNBBBNBBNBBCBHCBHHNHCBBCBHCB")
def test_load_map(self): expansions = load_map(get_test_file_path("samples/d14.txt")) expected = { "CH": "B", "HH": "N", "CB": "H", "NH": "C", "HB": "C", "HC": "B", "HN": "C", "NN": "C", "BH": "H", "NC": "B", "NB": "B", "BN": "B", "BB": "N", "BC": "B", "CC": "N", "CN": "C" } self.assertDictEqual(expansions, expected)
def test_encoded_expand(self): expansions = load_map(get_test_file_path("samples/d14.txt")) encoded_poly = encode_poly("NNCB") expected = encode_poly("NCNBCHB") expanded = expand_encoded(expansions, encoded_poly) self.assertDictEqual(expanded, expected)
def test_encoded_score(self): expansions = load_map(get_test_file_path("samples/d14.txt")) expanded = apply_steps(10, expansions, "NNCB") encoded = encode_poly(expanded) self.assertEqual(get_encoded_score(encoded), 1588)
def test_encoded_example(self): expansions = load_map(get_test_file_path("samples/d14.txt")) score = score_steps(10, expansions, "NNCB") self.assertEqual(score, 1588)
def test_example(self): expansions = load_map(get_test_file_path("samples/d14.txt")) expanded = apply_steps(10, expansions, "NNCB") self.assertEqual(len(expanded), 3073) self.assertEqual(get_score(expanded), 1588)